From 60c7e60828e1add0490e86434dc5e27e7a334397 Mon Sep 17 00:00:00 2001 From: Jaroslav Fowkes Date: Mon, 6 Jan 2025 09:29:23 +0000 Subject: [PATCH] Fix kernel regression test --- .../gradient_descent_kernel_test.py | 19 +++++++++---------- 1 file changed, 9 insertions(+), 10 deletions(-) diff --git a/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py b/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py index 9749f8e3..675d7690 100644 --- a/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py +++ b/test/accelerate_tests/base_tests/gradient_descent_kernel_test.py @@ -168,22 +168,21 @@ def test_make_a012_notb(self): [ 0., 0., 0.]]], dtype=FLOAT_TYPE) np.testing.assert_array_almost_equal(exp_A1, GDK.npy.LLerr, err_msg="`LLerr` buffer (=A1) has not been updated as expected") - exp_A2 = np.array([[[ 0., 4., 12.], - [ 4., 8., 16.], - [12., 16., 24.]], + exp_A2 = np.array([[[ 0., 2., 8.], + [ 2., 4., 10.], + [ 8., 10., 16.]], - [[ 0., 12., 28.], - [12., 24., 40.], - [28., 40., 56.]], + [[ 0., 2., 8.], + [ 2., 4., 10.], + [ 8., 10., 16.]], - [[ 0., 20., 44.], - [20., 40., 64.], - [44., 64., 88.]], + [[ 0., 2., 8.], + [ 2., 4., 10.], + [ 8., 10., 16.]], [[ 0., 0., 0.], [ 0., 0., 0.], [ 0., 0., 0.]]], dtype=FLOAT_TYPE) - print(GDK.npy.LLden) np.testing.assert_array_almost_equal(exp_A2, GDK.npy.LLden, err_msg="`LLden` buffer (=A2) has not been updated as expected")