提交 f3eb824d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

We don't support cudnn v3.

上级 362e09b3
...@@ -1313,8 +1313,6 @@ class GpuDnnSoftmaxBase(DnnBase): ...@@ -1313,8 +1313,6 @@ class GpuDnnSoftmaxBase(DnnBase):
DnnBase.__init__(self, [self.file], self.c_func) DnnBase.__init__(self, [self.file], self.c_func)
assert(algo in ('fast', 'accurate', 'log')) assert(algo in ('fast', 'accurate', 'log'))
if algo == 'log' and version(raises=False) < 3000:
raise RuntimeError("Need cuDNN v3 for log-softmax")
self.algo = algo self.algo = algo
assert(mode in ('instance', 'channel')) assert(mode in ('instance', 'channel'))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论