提交 b4b5740d authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #5666 from lamblin/last_before_rc3

Last before rc3
......@@ -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)
......
......@@ -3781,7 +3781,7 @@ class Split(Op):
return self.make_node(eval_points[0], *inputs[1:]).outputs
def c_code_cache_version(self):
return (1,)
return (2,)
def c_support_code(self):
return """
......@@ -3897,7 +3897,7 @@ class Split(Op):
ndim, split_dims,
%(x_typenum)s,
PyArray_STRIDES(%(x)s),
PyArray_DATA(%(x)s) + data_offset,
PyArray_BYTES(%(x)s) + data_offset,
%(x_itemsize)s,
PyArray_FLAGS(%(x)s),
NULL);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论