Fix unused variable in gpu device storage

apache · Feb 27, 2019 · 200d949 · 200d949
1 parent e3f609b
commit 200d949
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/storage/gpu_device_storage.h b/src/storage/gpu_device_storage.h
@@ -55,9 +55,9 @@ class GPUDeviceStorage {
 };  // class GPUDeviceStorage
 
 inline void* GPUDeviceStorage::Alloc(Storage::Handle* handle) {
-  const size_t size = handle->size;
   void* ret = nullptr;
 #if MXNET_USE_CUDA
+  const size_t size = handle->size;
   mxnet::common::cuda::DeviceStore device_store(handle->ctx.real_dev_id(), true);
 #if MXNET_USE_NCCL
   std::lock_guard<std::mutex> l(Storage::Get()->GetMutex(Context::kGPU));