From 9e0203196d4bcc3d8f41a11547828aebdd2a9547 Mon Sep 17 00:00:00 2001 From: Baptiste Nicolet Date: Tue, 19 Jul 2022 14:46:02 +0200 Subject: [PATCH] Updated test to reflect recent drjit API changes --- tests/test_cholesky.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_cholesky.py b/tests/test_cholesky.py index 91ff13f..fdc362b 100644 --- a/tests/test_cholesky.py +++ b/tests/test_cholesky.py @@ -186,7 +186,7 @@ def test_frameworks(framework): solver = CholeskySolverF(n_verts, drjit.cuda.TensorXi(idx[0]), drjit.cuda.TensorXi(idx[1]), drjit.cuda.TensorXf64(values), MatrixType.COO) b_drjit = drjit.cuda.TensorXf(b) - x_drjit = drjit.zero(drjit.cuda.TensorXf, b.shape) + x_drjit = drjit.zeros(drjit.cuda.TensorXf, b.shape) solver.solve(b_drjit, x_drjit) assert(np.allclose(x_drjit.numpy(), x_ref)) @@ -194,6 +194,6 @@ def test_frameworks(framework): solver = CholeskySolverF(n_verts, drjit.llvm.TensorXi(idx[0]), drjit.llvm.TensorXi(idx[1]), drjit.llvm.TensorXf64(values), MatrixType.COO) b_drjit = drjit.llvm.TensorXf(b) - x_drjit = drjit.zero(drjit.llvm.TensorXf, b.shape) + x_drjit = drjit.zeros(drjit.llvm.TensorXf, b.shape) solver.solve(b_drjit, x_drjit) assert(np.allclose(x_drjit.numpy(), x_ref))