diff options
author | Yangqing Jia <jiayq@google.com> | 2014-10-09 21:41:11 -0700 |
---|---|---|
committer | Yangqing Jia <jiayq@google.com> | 2014-10-10 09:04:14 -0700 |
commit | 44bc1552781eea958c228981a8638ddfcf80e3d9 (patch) | |
tree | 516dea7faf692285cf37aa97c92aeb1a11b44d6a /src/caffe/layers/cudnn_relu_layer.cu | |
parent | 7eecdf9c804e2b6a162e44c3a7a4b58317ed394b (diff) | |
download | caffeonacl-44bc1552781eea958c228981a8638ddfcf80e3d9.tar.gz caffeonacl-44bc1552781eea958c228981a8638ddfcf80e3d9.tar.bz2 caffeonacl-44bc1552781eea958c228981a8638ddfcf80e3d9.zip |
fix instantiation
Diffstat (limited to 'src/caffe/layers/cudnn_relu_layer.cu')
-rw-r--r-- | src/caffe/layers/cudnn_relu_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/cudnn_relu_layer.cu b/src/caffe/layers/cudnn_relu_layer.cu index e3900f37..86250870 100644 --- a/src/caffe/layers/cudnn_relu_layer.cu +++ b/src/caffe/layers/cudnn_relu_layer.cu @@ -45,7 +45,7 @@ void CuDNNReLULayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, this->bottom_desc_, bottom_data, this->bottom_desc_, bottom_diff)); } -INSTANTIATE_CLASS(CuDNNReLULayer); +INSTANTIATE_LAYER_GPU_FUNCS(CuDNNReLULayer); } // namespace caffe #endif |