提交 be54c3ac authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Remove max version of cuDNN

The predefined constants (enum) are the same with v6
上级 029e5d20
......@@ -1994,7 +1994,8 @@ class _RNNDescriptor(DnnBase):
hidden_size = as_i32(hidden_size)
num_layers = as_i32(num_layers)
assert 5000 < version() < 5200, "Constants only work for cudnn 5, 5.1"
if version() < 5005:
raise RuntimeError("cudnn RNN require cudnn v5 final or higher.")
if input_mode == 'linear':
input_mode = as_i32(0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论