From 9b6606504241f27a9d42b96f535bf5f2c2918161 Mon Sep 17 00:00:00 2001
From: Determinant <ted.sybil@gmail.com>
Date: Mon, 18 May 2015 23:57:12 +0800
Subject: add warning options to nvcc

---
 matrix/cukernel.cu | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'matrix')

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,
-- 
cgit v1.2.3