Merge pull request #996 from IntelPython/fix/kernel_pipeline_for_parfors

Diptorup Deb · web-flow · commit bbf978e3be87 · 2023-04-12T10:34:15.000-05:00
Use kernel compiler for parfors as well.
diff --git a/numba_dpex/core/compiler.py b/numba_dpex/core/compiler.py
@@ -85,7 +85,7 @@ def compile_with_dpex(
             return_type=return_type,
             flags=flags,
             locals={},
-            pipeline_class=OffloadCompiler,
+            pipeline_class=KernelCompiler,
         )
     else:
         raise UnreachableError()

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ def compile_with_dpex(`
`85`	`85`	`return_type=return_type,`
`86`	`86`	`flags=flags,`
`87`	`87`	`locals={},`
`88`		`- pipeline_class=OffloadCompiler,`
	`88`	`+ pipeline_class=KernelCompiler,`
`89`	`89`	`)`
`90`	`90`	`else:`
`91`	`91`	`raise UnreachableError()`