Skip to content
Snippets Groups Projects
Commit b4b105be authored by Richard Angersbach's avatar Richard Angersbach
Browse files

Minor fix

parent a9590829
No related branches found
No related tags found
1 merge request!438Reduction Support
Pipeline #73883 failed
...@@ -31,8 +31,8 @@ def test_reduction(target, dtype, op): ...@@ -31,8 +31,8 @@ def test_reduction(target, dtype, op):
vectorize_info = {'instruction_set': 'avx', 'assume_inner_stride_one': True} vectorize_info = {'instruction_set': 'avx', 'assume_inner_stride_one': True}
config = ps.CreateKernelConfig(target=ps.Target.GPU) if gpu_avail \ config = ps.CreateKernelConfig(target=target) if gpu_avail \
else ps.CreateKernelConfig(cpu_openmp=True, cpu_vectorize_info=vectorize_info) else ps.CreateKernelConfig(target=target, cpu_openmp=True, cpu_vectorize_info=vectorize_info)
ast_reduction = ps.create_kernel([red_assign], config, default_dtype=dtype) ast_reduction = ps.create_kernel([red_assign], config, default_dtype=dtype)
ps.show_code(ast_reduction) ps.show_code(ast_reduction)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment