fix problem when compiling without CUDA

dmlc · Jun 9, 2015 · 1508863 · 1508863
1 parent f3d68ee
commit 1508863
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/apps/ps/mnist_mlp_ps.cpp b/apps/ps/mnist_mlp_ps.cpp
@@ -208,7 +208,9 @@ int MinervaWorkerMain(int rank, int size, int argc, char *argv[]) {
         bias[k] = NArray::PushGradAndPullWeight(sens[k + 1].Sum(1) / mb_size, GetBiasName(k));
         weights[k] = NArray::PushGradAndPullWeight(sens[k + 1] * acts[k].Trans() / mb_size, GetWeightName(k));
       }
+#ifdef HAS_CUDA
       ms.SetDevice(gpuDevice);
+#endif
 
       if ((mb - rank) % 20 == 0) {
         ms.SetDevice(cpuDevice);