From b4b105be0ff674bd1cacaea75e5f92bf8a7fda3c Mon Sep 17 00:00:00 2001 From: zy69guqi <richard.angersbach@fau.de> Date: Tue, 11 Feb 2025 17:21:54 +0100 Subject: [PATCH] Minor fix --- tests/kernelcreation/test_reduction.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/kernelcreation/test_reduction.py b/tests/kernelcreation/test_reduction.py index 67a844821..12dc4ba1c 100644 --- a/tests/kernelcreation/test_reduction.py +++ b/tests/kernelcreation/test_reduction.py @@ -31,8 +31,8 @@ def test_reduction(target, dtype, op): vectorize_info = {'instruction_set': 'avx', 'assume_inner_stride_one': True} - config = ps.CreateKernelConfig(target=ps.Target.GPU) if gpu_avail \ - else ps.CreateKernelConfig(cpu_openmp=True, cpu_vectorize_info=vectorize_info) + config = ps.CreateKernelConfig(target=target) if gpu_avail \ + else ps.CreateKernelConfig(target=target, cpu_openmp=True, cpu_vectorize_info=vectorize_info) ast_reduction = ps.create_kernel([red_assign], config, default_dtype=dtype) ps.show_code(ast_reduction) -- GitLab