[storage] use if

apache · Aug 11, 2015 · 21674ef · 21674ef
1 parent b2bc69f
commit 21674ef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/include/mxnet/cuda_utils.h b/include/mxnet/cuda_utils.h
@@ -8,7 +8,7 @@
 
 #include <dmlc/logging.h>
 
-#ifdef MXNET_USE_CUDA
+#if MXNET_USE_CUDA
 
 #include <cuda_runtime.h>
 #include <cublas_v2.h>
@@ -100,7 +100,7 @@ inline const char* CurandGetErrorString(curandStatus_t status) {
 
 #endif  // MXNET_USE_CUDA
 
-#ifdef MXNET_USE_CUDNN
+#if MXNET_USE_CUDNN
 
 #include <cudnn.h>
 

diff --git a/src/storage/gpu_storage.cc b/src/storage/gpu_storage.cc
@@ -3,7 +3,7 @@
  */
 #include "./gpu_storage.h"
 #include "mxnet/cuda_utils.h"
-#ifdef MXNET_USE_CUDA
+#if MXNET_USE_CUDA
 #include <cuda_runtime.h>
 #endif  // MXNET_USE_CUDA
 
@@ -12,7 +12,7 @@ namespace storage {
 
 void* GpuStorage::Alloc(size_t size) {
   void* ret;
-#ifdef MXNET_USE_CUDA
+#if MXNET_USE_CUDA
   CUDA_CALL(cudaMalloc(&ret, size));
 #else   // MXNET_USE_CUDA
   LOG(FATAL) << "Please compile with CUDA enabled";
@@ -21,7 +21,7 @@ void* GpuStorage::Alloc(size_t size) {
 }
 
 void GpuStorage::Free(void* ptr) {
-#ifdef MXNET_USE_CUDA
+#if MXNET_USE_CUDA
   CUDA_CALL(cudaFree(ptr));
 #else   // MXNET_USE_CUDA
   LOG(FATAL) << "Please compile with CUDA enabled";