Added preprocessor flags for mac and windows support

soumith · Aug 18, 2014 · 119c2a1 · 119c2a1
1 parent ad90c9e
commit 119c2a1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/cudaconv3/src/img_acts.cu b/cudaconv3/src/img_acts.cu
@@ -4,6 +4,10 @@
 
 #include "../include/cudaconv2.cuh"
 
+#if defined(_WIN64) || defined(_WIN32) || defined(__APPLE__)
+#define uint unsigned int
+#endif
+
 /*
  * Block size: 16x16.
  * blockIdx.x determines case in batches of 16*imgsPerThread.

diff --git a/nvmatrix/include/nvmatrix_kernels.cuh b/nvmatrix/include/nvmatrix_kernels.cuh
@@ -19,7 +19,7 @@
 
 #include <curand_kernel.h>
 
-#if defined(_WIN64) || defined(_WIN32)
+#if defined(_WIN64) || defined(_WIN32) || defined(__APPLE__)
 #define uint unsigned int
 #endif