提交 c987822b authored 作者: Colin Raffel's avatar Colin Raffel

Changing vector to fvector to force 32 bit

上级 c0d3bff3
...@@ -294,7 +294,7 @@ def test_local_gpu_subtensor(): ...@@ -294,7 +294,7 @@ def test_local_gpu_subtensor():
def test_local_split(): def test_local_split():
""" Test that the GpuSplit op is being applied and works """ """ Test that the GpuSplit op is being applied and works """
# Construct symbolic split # Construct symbolic split
x = tensor.vector() x = tensor.fvector()
splits = tensor.lvector() splits = tensor.lvector()
ra, rb, rc = tensor.split(x, splits, n_splits=3, axis=0) ra, rb, rc = tensor.split(x, splits, n_splits=3, axis=0)
# Compile function to use CPU # Compile function to use CPU
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论