Fix warning when CPU only (apache#4751)

rravu3 · Jan 21, 2017 · da8cbe6 · da8cbe6
1 parent 59adf6f
commit da8cbe6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/engine/profiler.cc b/src/engine/profiler.cc
@@ -27,11 +27,13 @@ Profiler::Profiler()
   this->init_time_ = NowInUsec();
 
   // TODO(ziheng) get device number during execution
-  int kMaxNumCpus = 64, kMaxNumGpus = 32;
+  int kMaxNumCpus = 64;
   this->cpu_num_ = kMaxNumCpus;
-  this->gpu_num_ = 0;
 #if MXNET_USE_CUDA
+  int kMaxNumGpus = 32;
   this->gpu_num_ = kMaxNumGpus;
+#else
+  this->gpu_num_ = 0;
 #endif
 
   this->profile_stat = new DevStat[cpu_num_ + gpu_num_ + 1];