diff options
Diffstat (limited to 'mlir/test/python/ir')
| -rw-r--r-- | mlir/test/python/ir/array_attributes.py | 22 | ||||
| -rw-r--r-- | mlir/test/python/ir/operation.py | 4 |
2 files changed, 24 insertions, 2 deletions
diff --git a/mlir/test/python/ir/array_attributes.py b/mlir/test/python/ir/array_attributes.py index ef1d835..66f7ec8 100644 --- a/mlir/test/python/ir/array_attributes.py +++ b/mlir/test/python/ir/array_attributes.py @@ -31,6 +31,7 @@ def testGetDenseElementsUnsupported(): # CHECK: unimplemented array format conversion from format: print(e) + # CHECK-LABEL: TEST: testGetDenseElementsUnSupportedTypeOkIfExplicitTypeProvided @run def testGetDenseElementsUnSupportedTypeOkIfExplicitTypeProvided(): @@ -41,8 +42,9 @@ def testGetDenseElementsUnSupportedTypeOkIfExplicitTypeProvided(): # realistic example would be a NumPy extension type like the bfloat16 # type from the ml_dtypes package, which isn't a dependency of this # test. - attr = DenseElementsAttr.get(array.view(np.datetime64), - type=IntegerType.get_signless(64)) + attr = DenseElementsAttr.get( + array.view(np.datetime64), type=IntegerType.get_signless(64) + ) # CHECK: dense<{{\[}}[1, 2, 3], [4, 5, 6]]> : tensor<2x3xi64> print(attr) # CHECK: {{\[}}[1 2 3] @@ -135,6 +137,7 @@ def testGetDenseElementsFromListMixedTypes(): # Splats. ################################################################################ + # CHECK-LABEL: TEST: testGetDenseElementsSplatInt @run def testGetDenseElementsSplatInt(): @@ -617,3 +620,18 @@ def testGetDenseResourceElementsAttr(): # CHECK: BACKING MEMORY DELETED # CHECK: EXIT FUNCTION print("EXIT FUNCTION") + + +# CHECK-LABEL: TEST: testDanglingResource +print("TEST: testDanglingResource") +# see https://github.com/llvm/llvm-project/pull/149414, https://github.com/llvm/llvm-project/pull/150137, https://github.com/llvm/llvm-project/pull/150561 +# This error occurs only when there is an alive context with a DenseResourceElementsAttr +# in the end of the program, so we put it here without an encapsulating function. +ctx = Context() + +with ctx, Location.unknown(): + DenseResourceElementsAttr.get_from_buffer( + memoryview(np.array([1, 2, 3])), + "some_resource", + RankedTensorType.get((3,), IntegerType.get_signed(32)), + ) diff --git a/mlir/test/python/ir/operation.py b/mlir/test/python/ir/operation.py index ede1571..c6b5daf 100644 --- a/mlir/test/python/ir/operation.py +++ b/mlir/test/python/ir/operation.py @@ -978,8 +978,12 @@ def testModuleMerge(): foo = m1.body.operations[0] bar = m2.body.operations[0] qux = m2.body.operations[1] + assert bar.is_before_in_block(qux) bar.move_before(foo) + assert bar.is_before_in_block(foo) qux.move_after(foo) + assert bar.is_before_in_block(qux) + assert foo.is_before_in_block(qux) # CHECK: module # CHECK: func private @bar |