提交 2389c490 authored 作者: --global's avatar --global

Update checks for self.workmem value in GpuDnnConv

上级 8c7ab092
...@@ -419,7 +419,7 @@ class GpuDnnConv(DnnBase, COp): ...@@ -419,7 +419,7 @@ class GpuDnnConv(DnnBase, COp):
if version() < (3000, 3000): if version() < (3000, 3000):
if self.workmem == 'fft': if self.workmem == 'fft':
raise RuntimeError("CuDNN FFT convolution requires CuDNN v3") raise RuntimeError("CuDNN FFT convolution requires CuDNN v3")
elif self.workmem == 'time': elif self.workmem in ['time', 'time_once']:
raise RuntimeError("CuDNN convolution timing requires CuDNN v3") raise RuntimeError("CuDNN convolution timing requires CuDNN v3")
assert self.workmem in ['none', 'small', 'large', 'fft', 'time', assert self.workmem in ['none', 'small', 'large', 'fft', 'time',
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论