From 23685324b94ef1248ad5b377121f4e71cda24293 Mon Sep 17 00:00:00 2001
From: gdh1995 <gdh1995@qq.com>
Date: Sun, 26 Apr 2015 22:52:41 +0800
Subject: [PATCH] Net::Update: CPU_ONLY is in wrong place

---
 src/caffe/net.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/caffe/net.cpp b/src/caffe/net.cpp
index fd00b12..888eec1 100644
--- a/src/caffe/net.cpp
+++ b/src/caffe/net.cpp
@@ -756,15 +756,15 @@ void Net<Dtype>::Update() {
       owner_diff = params_[param_owners_[i]]->mutable_cpu_diff();
       caffe_add(count, this_diff, owner_diff, owner_diff);
       break;
-#ifndef CPU_ONLY
     case Caffe::GPU:
+#ifndef CPU_ONLY
       this_diff = params_[i]->gpu_diff();
       owner_diff = params_[param_owners_[i]]->mutable_gpu_diff();
       caffe_gpu_add(count, this_diff, owner_diff, owner_diff);
-      break;
 #else
       NO_GPU;
 #endif
+      break;
     default:
       LOG(FATAL) << "Unknown caffe mode: " << Caffe::mode();
     }
-- 
2.7.4