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