Skip to content

Commit 73f5a95

Browse files
authored
Fix copy semantics (#78)
1 parent 1627eb3 commit 73f5a95

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

imex_version.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2875ae696359f267571a77ea774d81659a10d667
1+
149d88b36ed4f7ff19fec3e3d9842aa6dcd4fb3d

src/jit/mlir.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -550,6 +550,7 @@ static const char *cpu_pipeline = "ndarray-dist,"
550550
"imex-remove-temporaries,"
551551
"func.func(convert-linalg-to-parallel-loops),"
552552
"func.func(scf-parallel-loop-fusion),"
553+
"drop-regions,"
553554
"canonicalize,"
554555
"fold-memref-alias-ops,"
555556
"expand-strided-metadata,"
@@ -602,7 +603,7 @@ static const char *gpu_pipeline =
602603
"func.func(convert-parallel-loops-to-gpu),"
603604
// insert-gpu-allocs pass can have client-api = opencl or vulkan args
604605
"func.func(insert-gpu-allocs{in-regions=1}),"
605-
"convert-region-to-gpu,"
606+
"drop-regions,"
606607
"canonicalize,"
607608
"normalize-memrefs,"
608609
// Unstride memrefs does not seem to be needed.

test/test_manip.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,13 +69,12 @@ def test_astype_suite(self):
6969
assert b.dtype == to_type
7070
assert numpy.allclose(sp.to_numpy(b), [0, 1, 2, 3, 4, 5, 6, 7])
7171

72-
@pytest.mark.skip(reason="FIXME copy is eliminated")
7372
def test_todevice_host2host(self):
7473
a = sp.arange(0, 8, 1, sp.int32)
7574
b = a.to_device()
7675
assert numpy.allclose(sp.to_numpy(b), [0, 1, 2, 3, 4, 5, 6, 7])
7776

78-
@pytest.mark.skip(reason="FIXME copy is eliminated, device support")
77+
@pytest.mark.skip(reason="device support")
7978
def test_todevice_host2gpu(self):
8079
a = sp.arange(0, 8, 1, sp.int32)
8180
b = a.to_device(device="GPU")

0 commit comments

Comments
 (0)