Merge pull request #64 from abergeron/nan

Add definition of NAN to the cluda blurb.
Theano · Aug 3, 2015 · 172cf4a · 172cf4a
2 parents 69e6052 + 743e8a3
commit 172cf4a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/gpuarray_buffer_cuda.c b/src/gpuarray_buffer_cuda.c
@@ -180,6 +180,8 @@ static const char CUDA_PREAMBLE[] =
     "#define LOCAL_MEM __shared__\n"
     "#define LOCAL_MEM_ARG /* empty */\n"
     "#define REQD_WG_SIZE(X,Y,Z) __launch_bounds__(X*Y, Z)\n"
+    "#include <math_constants.h>\n"
+    "#define NAN CUDART_NAN_F\n"
     "#define LID_0 threadIdx.x\n"
     "#define LID_1 threadIdx.y\n"
     "#define LID_2 threadIdx.z\n"