提交 32b75e7f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

v5rc has different prototypes for some versions so reject it.

上级 aab7cead
...@@ -1874,6 +1874,8 @@ class _RNNDescriptor(DnnBase): ...@@ -1874,6 +1874,8 @@ class _RNNDescriptor(DnnBase):
__props__ = ('context_name',) __props__ = ('context_name',)
def __init__(self, context_name): def __init__(self, context_name):
if version() < 5005:
raise RuntimeError("cudnn RNN require cudnn v5 final or higher.")
DnnBase.__init__(self, ["dnn_rnn_desc.c"], "dnn_rnn_desc") DnnBase.__init__(self, ["dnn_rnn_desc.c"], "dnn_rnn_desc")
self.context_name = context_name self.context_name = context_name
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论