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

Fix condition based on CuDNN version

上级 6f1dc3d8
...@@ -672,7 +672,7 @@ class GpuDnnConvGradI(DnnBase): ...@@ -672,7 +672,7 @@ class GpuDnnConvGradI(DnnBase):
# The small-workspace implementation is only available from CuDNN V4 # The small-workspace implementation is only available from CuDNN V4
# onward. # onward.
if version() < (4000, 4000) and self.algo == 'fft_tiling': if version() < 4000 and self.algo == 'fft_tiling':
raise RuntimeError("CuDNN's tiled-FFT convolution requires CuDNN " raise RuntimeError("CuDNN's tiled-FFT convolution requires CuDNN "
"v4 or more recent") "v4 or more recent")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论