From ce18be44cb35bc89ba0920623410f06219d49ef0 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Thu, 18 Jul 2013 16:52:20 -0400 Subject: [PATCH] Test sliced reduction --- test/test_algorithm.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/test/test_algorithm.py b/test/test_algorithm.py index fc31e8c5..59fa60f0 100644 --- a/test/test_algorithm.py +++ b/test/test_algorithm.py @@ -245,10 +245,16 @@ def test_sum(ctx_factory): a = a_gpu.get() - sum_a = np.sum(a) - sum_a_gpu = cl_array.sum(a_gpu).get() - - assert abs(sum_a_gpu - sum_a) / abs(sum_a) < 1e-4 + for slc in [ + slice(None), + slice(1000, 3000), + slice(1000, -3000), + slice(1000, None), + ]: + sum_a = np.sum(a[slc]) + sum_a_gpu = cl_array.sum(a_gpu[slc]).get() + + assert abs(sum_a_gpu - sum_a) / abs(sum_a) < 1e-4 @pytools.test.mark_test.opencl -- GitLab