Update tensor_gpu-inl.cuh

stefanhenneking · Nov 21, 2014 · c87fa59 · c87fa59
1 parent ef1f300
commit c87fa59
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mshadow/cuda/tensor_gpu-inl.cuh b/mshadow/cuda/tensor_gpu-inl.cuh
@@ -10,8 +10,11 @@
 
 namespace mshadow{
     namespace cuda{
+        #ifndef __CUDA_ARCH__
+        #warning "__CUDA_ARCH__ is not defined, I will assume compiling with CUDA verion greater than 2.0"
+        #endif
         /* load unit for memory access */
-        #if __CUDA_ARCH__>=200
+        #if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 200
         const int kMemUnitBits = 5;
         const int kMaxThreadsPerBlock = 1024;
         #else