Feat C++ API
A feature engineering automation tool
cuda_utils.h
Go to the documentation of this file.
1
/* FEAT
2
copyright 2017 William La Cava
3
license: GNU/GPL v3
4
*/
5
#ifndef CUDA_UTILS_H
6
#define CUDA_UTILS_H
7
8
9
#ifdef _OPENMP
10
#include <omp.h>
11
#else
12
#define omp_get_thread_num() 0
13
#define omp_get_max_threads() 1
14
#endif
15
16
extern
int
NUM_SMS
;
17
extern
int
DIM_GRID
;
18
extern
int
DIM_BLOCK
;
19
20
21
namespace
FT
{
22
23
namespace
Pop{
24
namespace
Op{
25
26
void
choose_gpu
();
27
28
void
initialize_cuda
();
29
30
}
31
}
32
}
33
34
#endif
DIM_GRID
int DIM_GRID
NUM_SMS
int NUM_SMS
DIM_BLOCK
int DIM_BLOCK
FT::Pop::Op::initialize_cuda
void initialize_cuda()
FT::Pop::Op::choose_gpu
void choose_gpu()
FT
main Feat namespace
Definition:
data.cc:13
src
pop
cuda-op
cuda_utils.h
Generated by
1.9.1