提交 8fbc2d0a authored 作者: Caglar's avatar Caglar

fixed the problem related to the cudandarray and inplace assignment

上级 ce066d1d
......@@ -91,7 +91,7 @@ class GpuSolve(GpuOp):
cula.culaDeviceSgesv(n, nrhs, A_ptr, lda, ipiv_ptr, b_ptr, ldb)
return A, b
A, b = cula_gpu_solve(A_pycuda, b_pycuda)
A_pycuda, b_pycuda = cula_gpu_solve(A_pycuda, b_pycuda)
z[0] = b
thunk.inputs = inputs
......
......@@ -35,7 +35,7 @@ class TestCula(unittest.TestCase):
solver = cula.gpu_solve(A, b)
fn = theano.function([A, b], [solver])
res = fn(A_val, b_val)
res[0].get(x_res)
x_res = numpy.array(res[0])
utt.assert_allclose(x_res, x_val)
def test_diag_solve(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论