Join GitHub today
GitHub is home to over 20 million developers working together to host and review code, manage projects, and build software together.
Basic cuDNN v3 support (update) #3160
Conversation
shelhamer
added the
in progress
label
Oct 6, 2015
|
@shelhamer Regarding the test failure for Groups -- it seems like this->weight_offset_ for the cuDNN routines is getting set incorrectly / not set (either way, it's wrong!) This seems to have been introduced in 9d8206e Setting it back explicitly to: in CuDNNConvolutionLayer::Setup seems to fix the issue, not sure if there's a better way - let me know and I'll update the PR |
shelhamer
added ready for review speed-up and removed in progress
labels
Oct 16, 2015
shelhamer
added a commit
that referenced
this pull request
Oct 16, 2015
|
|
shelhamer |
321720d
|
shelhamer
merged commit 321720d
into
BVLC:master
Oct 16, 2015
1 check passed
shelhamer
deleted the
shelhamer:cudnnV3 branch
Oct 16, 2015
|
Great |
|
cuDNN v3 is not itself backward compatible with v2, so adopting v3 in this PR does deprecate v2. We plan to follow the latest cuDNN version in master but keep compatability as the cuDNN interface itself allows. |
This was referenced Oct 19, 2015
ronghanghu
commented on the diff
Oct 28, 2015
| } | ||
| Dtype* bias_diff = NULL; | ||
| if (this->bias_term_ && this->param_propagate_down_[1]) { | ||
| bias_diff = this->blobs_[1]->mutable_gpu_diff(); | ||
| + caffe_gpu_set(this->blobs_[1]->count(), Dtype(0), bias_diff); |
ronghanghu
Member
|
shelhamer commentedOct 6, 2015
This is the same as #2737 except for
cuDNNConvolutionLayer::Forward_gpu()now that algo and workspace are determined inReshape().Thanks @slayton58 for the integration.