aboutsummaryrefslogtreecommitdiff
path: root/matrix
diff options
context:
space:
mode:
authorDeterminant <[email protected]>2015-05-18 23:57:12 +0800
committerDeterminant <[email protected]>2015-05-18 23:57:12 +0800
commit9b6606504241f27a9d42b96f535bf5f2c2918161 (patch)
treef14cbde57446d94e7a45ea564d996157cb88d4f0 /matrix
parent186cf4f39e1c753a6056101f654d2939f812d285 (diff)
add warning options to nvcc
Diffstat (limited to 'matrix')
-rw-r--r--matrix/cukernel.cu7
1 files changed, 4 insertions, 3 deletions
diff --git a/matrix/cukernel.cu b/matrix/cukernel.cu
index dd1ebfc..ee6d871 100644
--- a/matrix/cukernel.cu
+++ b/matrix/cukernel.cu
@@ -84,6 +84,7 @@ __global__ void block_reduce_max(const float *input, float *output,
}
extern "C" {
+#include "cukernel.h"
void cuda_sigmoid(const Matrix *a, Matrix *b) {
dim3 threadsPerBlock(CUDA_THREADS_N,
CUDA_THREADS_N);
@@ -106,7 +107,7 @@ extern "C" {
a->stride / sizeof(float), stride / sizeof(float),
ncol);
ncol = blocks_per_row;
- assert(ncol <= block.x);
+ assert((unsigned long)ncol <= block.x);
grid.x = 1;
block_reduce_sum<<<grid, block, block.x * sizeof(float)>>> \
(res, b->data.f,
@@ -143,7 +144,7 @@ extern "C" {
max->stride / sizeof(float),
ncol);
ncol = blocks_per_row;
- assert(ncol <= block.x);
+ assert((unsigned long)ncol <= block.x);
grid.x = 1;
block_reduce_sum<<<grid, block, block.x * sizeof(float)>>> \
(res, b->data.f,
@@ -165,7 +166,7 @@ extern "C" {
a->stride / sizeof(float), stride / sizeof(float),
ncol);
ncol = blocks_per_row;
- assert(ncol <= block.x);
+ assert((unsigned long)ncol <= block.x);
grid.x = 1;
block_reduce_max<<<grid, block, block.x * sizeof(float)>>> \
(res, b->data.f,