diff options
Diffstat (limited to 'ot/gpu/cudamat/cudamat/learn_kernels.cuh')
-rw-r--r-- | ot/gpu/cudamat/cudamat/learn_kernels.cuh | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/ot/gpu/cudamat/cudamat/learn_kernels.cuh b/ot/gpu/cudamat/cudamat/learn_kernels.cuh new file mode 100644 index 0000000..26c8abd --- /dev/null +++ b/ot/gpu/cudamat/cudamat/learn_kernels.cuh @@ -0,0 +1,12 @@ +#ifndef EBM_KERNELS_H_ +#define EBM_KERNELS_H_ + +#define NUM_VECTOR_OP_BLOCKS 4096 +#define NUM_VECTOR_OP_THREADS_PER_BLOCK 512 + +#define NUM_SPARSE_GRAD_BLOCKS 4096 +#define NUM_SPARSE_GRAD_THREADS_PER_BLOCK 512 + +__global__ void kMultiplyBySigmoidGrad(double* act, double* target, const unsigned int len); + +#endif |