From 3cc9b73a011b7ba8964d1bccc97b944814cbeb38 Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Tue, 24 Mar 2015 14:17:57 -0700 Subject: [PATCH] note cuDNN v2 convolutional TODOs --- src/caffe/layers/cudnn_conv_layer.cu | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/caffe/layers/cudnn_conv_layer.cu b/src/caffe/layers/cudnn_conv_layer.cu index ed0d91db9d7..08f5201bc22 100644 --- a/src/caffe/layers/cudnn_conv_layer.cu +++ b/src/caffe/layers/cudnn_conv_layer.cu @@ -23,7 +23,10 @@ void CuDNNConvolutionLayer::Forward_gpu( for (int g = 0; g < this->group_; g++) { cudnnConvolutionFwdAlgo_t algo; - // get the desired convolution algorithm + // pick the convolution algorithm + // TODO(shelhamer) this should be done during reshape + // TODO(shelhamer) the choice of automatic or manual algorithm picking + // should be exposed in proto CUDNN_CHECK(cudnnGetConvolutionForwardAlgorithm(handle_[g], bottom_descs_[i], filter_desc_,