WIP...

IntelPython · May 31, 2023 · 4af7777 · 4af7777
1 parent 2e38579
commit 4af7777
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/driver.py b/driver.py
@@ -18,6 +18,12 @@ def f(a, b):
     return
 
 
+@dpex.dpjit
+def g(a):
+    c = dpnp.empty(a.shape, dtype=a.dtype)
+    return c
+
+
 m = 8
 n = 8
 a = dpnp.ones((m, n))
@@ -26,3 +32,6 @@ def f(a, b):
 f(a, b)
 
 print(b)
+
+c = g(b)
+print(c)
diff --git a/numba_dpex/core/passes/parfor_legalize_cfd_pass.py b/numba_dpex/core/passes/parfor_legalize_cfd_pass.py
@@ -316,7 +316,7 @@ def run_pass(self, state):
         """
         # Ensure we have an IR and type information.
         assert state.func_ir
-        # cfd_legalizer = ParforLegalizeCFDPassImpl(state)
-        # cfd_legalizer.run()
+        cfd_legalizer = ParforLegalizeCFDPassImpl(state)
+        cfd_legalizer.run()
 
         return True