提交 c89d22ff authored 作者: wonghang's avatar wonghang

Merge branch 'potrf64_and_Lop' of github.com:wonghang/Theano into potrf64_and_Lop

...@@ -1821,4 +1821,4 @@ KERNEL void tri(GLOBAL_MEM %(ctype)s *a, ga_size a_off, ...@@ -1821,4 +1821,4 @@ KERNEL void tri(GLOBAL_MEM %(ctype)s *a, ga_size a_off,
return s return s
def c_code_cache_version(self): def c_code_cache_version(self):
return (10,) return (1,)
...@@ -573,7 +573,6 @@ def test_gputri(): ...@@ -573,7 +573,6 @@ def test_gputri():
out, out,
mode=mode_with_gpu) mode=mode_with_gpu)
result = np.asarray(f(N, M, k)) - np.array(1).astype(dtype) result = np.asarray(f(N, M, k)) - np.array(1).astype(dtype)
print(result)
assert np.allclose(result, np.tri(N, M_, k, dtype=dtype)) assert np.allclose(result, np.tri(N, M_, k, dtype=dtype))
assert result.dtype == np.dtype(dtype) assert result.dtype == np.dtype(dtype)
assert any([isinstance(node.op, GpuTri) assert any([isinstance(node.op, GpuTri)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论