Merge pull request BVLC#1656 from longjon/fix-softmax-loss-stubs

Fix an error made in removing the SoftmaxLossLayer GPU stubs
longjon · Dec 30, 2014 · 9b3b440 · 9b3b440
2 parents a03d4f7 + 11add5b
commit 9b3b440
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/caffe/layers/softmax_loss_layer.cpp b/src/caffe/layers/softmax_loss_layer.cpp
@@ -85,11 +85,7 @@ void SoftmaxWithLossLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top,
   }
 }
 
-
-#ifdef CPU_ONLY
-STUB_GPU(SoftmaxWithLossLayer);
-#endif
-
 INSTANTIATE_CLASS(SoftmaxWithLossLayer);
 REGISTER_LAYER_CLASS(SOFTMAX_LOSS, SoftmaxWithLossLayer);
+
 }  // namespace caffe