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 
29 
30  }
31  }
32 }
33 
34 #endif
int DIM_GRID
int NUM_SMS
int DIM_BLOCK
void initialize_cuda()
void choose_gpu()
main Feat namespace
Definition: data.cc:13