commit 8cf718fd2b8c4df9076fcd3155740c2b25eccb1d
author Yori Zwols <yori@google.com> Tue May 26 15:12:07 2015 +0100
committer Yori Zwols <yori@google.com> Tue May 26 15:12:07 2015 +0100
tree cc7741f1f1b4397391e3b1d2af745ed243e3ff59
parent e014a57910f50e931f6765973e0f28f3f6d5beaa

Added compile-time flag to disable GPU checks

THCTensor.c

diff --git a/THCTensor.c b/THCTensor.c
index 8f0a5a7..b32948d 100644
--- a/THCTensor.c
+++ b/THCTensor.c
@@ -759,6 +759,9 @@
 
 int THCudaTensor_checkGPU(THCState *state, unsigned int nTensors, ...)
 {
+#ifdef DISABLE_CHECK_GPU
+  return 1;  // Disable GPU checks.
+#else
   int curDev = -1;
   THCudaCheck(cudaGetDevice(&curDev));
   va_list(args);
@@ -777,4 +780,5 @@
   }
   va_end(args);
   return valid;
+#endif
 }