diff --git a/tests/kernelcreation/test_reduction.py b/tests/kernelcreation/test_reduction.py index 67a844821e1323ed49f2dda5cb40d17a10c4908a..12dc4ba1cdb9f0a84562fc7e89d4a1dff68d5939 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)