提交 adbcbfc2 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3251 from f0k/fix-cudnn-copypaste-bug

Fix copy/paste bug for cuDNN time_on_shape_change mode
......@@ -781,7 +781,7 @@ class GpuDnnConvGradW(DnnBase, COp):
choose_alg = '1'
if self.algo == 'guess_once':
choose_alg_once = '1'
elif self.algo in ['time_once', 'guess_on_shape_change']:
elif self.algo in ['time_once', 'time_on_shape_change']:
# The convolution implementation should be chosen according
# to timing
alg = 'CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0'
......@@ -985,7 +985,7 @@ class GpuDnnConvGradI(DnnBase, COp):
choose_alg = '1'
if self.algo == 'guess_once':
choose_alg_once = '1'
elif self.algo in ['time_once', 'guess_on_shape_change']:
elif self.algo in ['time_once', 'time_on_shape_change']:
# The convolution implementation should be chosen according
# to timing
alg = 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_0'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论