remove SoftmaxLossLayer CPU_ONLY stubs, since there is no GPU version
authorJonathan L Long <jonlong@cs.berkeley.edu>
Tue, 30 Dec 2014 01:59:07 +0000 (17:59 -0800)
committerJonathan L Long <jonlong@cs.berkeley.edu>
Tue, 30 Dec 2014 01:59:07 +0000 (17:59 -0800)
src/caffe/layers/softmax_loss_layer.cpp

index 82c62f6..db8dd8b 100644 (file)
@@ -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