提交 fc7d7277 authored 作者: carriepl's avatar carriepl 提交者: Frederic

fix typos

上级 e200ca29
...@@ -483,7 +483,7 @@ class GpuDnnConv(DnnBase, COp): ...@@ -483,7 +483,7 @@ class GpuDnnConv(DnnBase, COp):
elif self.algo == 'fft': elif self.algo == 'fft':
# need v3 # need v3
alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT' alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT'
elif self.algo == 'fft_tiliing': elif self.algo == 'fft_tiling':
# need v4 # need v4
alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING' alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING'
elif self.algo in ['guess_once', 'guess_on_shape_change']: elif self.algo in ['guess_once', 'guess_on_shape_change']:
...@@ -949,7 +949,7 @@ class GpuDnnConvGradI(DnnBase, COp): ...@@ -949,7 +949,7 @@ class GpuDnnConvGradI(DnnBase, COp):
elif self.algo == 'fft': elif self.algo == 'fft':
# need v3, big workspace # need v3, big workspace
alg = 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT' alg = 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT'
elif self.algo == 'fft': elif self.algo == 'fft_tiling':
# need v4, big workspace, but less then fft # need v4, big workspace, but less then fft
alg = 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT_TILING' alg = 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_FFT_TILING'
elif self.algo in ['guess_once', 'guess_on_shape_change']: elif self.algo in ['guess_once', 'guess_on_shape_change']:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论