Merge pull request #11 from IntelPython/one-shot

Use one-shot-bufferize pass

Author: tkarna

imex_version.txt

@@ -1 +1 @@
-a02f09350a8eba081c92a7d0117334eb56c9fb5a
+5a7bb80ede5fe4fa8d56ee0dd77c4e5c1327fe09

src/jit/mlir.cpp

@@ -687,20 +687,11 @@ static const std::string cpu_pipeline =
     // "convert-shape-to-std,"
     "arith-expand,"
     "memref-expand,"
-    "arith-bufferize,"
-    // "func-bufferize,"
     "func.func(empty-tensor-to-alloc-tensor),"
-    "func.func(scf-bufferize),"
-    "func.func(tensor-bufferize),"
-    "func.func(bufferization-bufferize),"
-    "func.func(linalg-bufferize),"
-    "func.func(linalg-detensorize),"
-    "func.func(tensor-bufferize),"
-    "region-bufferize,"
+    "one-shot-bufferize,"
     "canonicalize,"
-    "func.func(finalizing-bufferize),"
-    "func.func(buffer-deallocation),"
     "imex-remove-temporaries,"
+    "func.func(buffer-deallocation),"
     "func.func(convert-linalg-to-parallel-loops),"
     "func.func(scf-parallel-loop-fusion),"
     "drop-regions,"

test/test_manip.py

@@ -26,6 +26,13 @@ def doit(aapi, **kwargs):
 
         assert runAndCompare(doit)
 
+    def test_reshape_copy(self):
+        a = sp.arange(0, 8, 1, sp.int32)
+        b = sp.reshape(a, [4, 2])
+        a[0] = 20
+        assert numpy.allclose(sp.to_numpy(a), [20, 1, 2, 3, 4, 5, 6, 7])
+        assert numpy.allclose(sp.to_numpy(b), [[0, 1], [2, 3], [4, 5], [6, 7]])
+
     @pytest.mark.skipif(len(device), reason="FIXME 64bit on GPU")
     def test_astype_f64i32(self):
         def doit(aapi, **kwargs):