diff options
Diffstat (limited to 'mlir/test/python/dialects')
16 files changed, 2 insertions, 16 deletions
diff --git a/mlir/test/python/dialects/gpu/module-to-binary-nvvm.py b/mlir/test/python/dialects/gpu/module-to-binary-nvvm.py index 96e3f6efcc5d..cda570fceb64 100644 --- a/mlir/test/python/dialects/gpu/module-to-binary-nvvm.py +++ b/mlir/test/python/dialects/gpu/module-to-binary-nvvm.py @@ -2,7 +2,7 @@ # RUN: %PYTHON %s | FileCheck %s from mlir.ir import * -import mlir.dialects.gpu as gpu +from mlir.dialects import gpu, llvm, nvvm import mlir.dialects.gpu.passes from mlir.passmanager import * diff --git a/mlir/test/python/dialects/gpu/module-to-binary-rocdl.py b/mlir/test/python/dialects/gpu/module-to-binary-rocdl.py index f5ee2856f395..6e8c13b8dea3 100644 --- a/mlir/test/python/dialects/gpu/module-to-binary-rocdl.py +++ b/mlir/test/python/dialects/gpu/module-to-binary-rocdl.py @@ -2,7 +2,7 @@ # RUN: %PYTHON %s | FileCheck %s from mlir.ir import * -import mlir.dialects.gpu as gpu +from mlir.dialects import gpu, llvm, rocdl import mlir.dialects.gpu.passes from mlir.passmanager import * diff --git a/mlir/test/python/dialects/transform.py b/mlir/test/python/dialects/transform.py index 3477539b8d2f..6c5e4e5505b1 100644 --- a/mlir/test/python/dialects/transform.py +++ b/mlir/test/python/dialects/transform.py @@ -12,7 +12,6 @@ def run(f): print("\nTEST:", f.__name__) f(module) print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_bufferization_ext.py b/mlir/test/python/dialects/transform_bufferization_ext.py index 39014735aa8f..03b16c324a40 100644 --- a/mlir/test/python/dialects/transform_bufferization_ext.py +++ b/mlir/test/python/dialects/transform_bufferization_ext.py @@ -13,7 +13,6 @@ def run(f): print("\nTEST:", f.__name__) f() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_debug_ext.py b/mlir/test/python/dialects/transform_debug_ext.py index 0826ed14d199..2dfdaed34386 100644 --- a/mlir/test/python/dialects/transform_debug_ext.py +++ b/mlir/test/python/dialects/transform_debug_ext.py @@ -19,7 +19,6 @@ def run(f): f(sequence.bodyTarget) transform.YieldOp() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_extras.py b/mlir/test/python/dialects/transform_extras.py index 3aeac681f8d4..c52d5b41e11b 100644 --- a/mlir/test/python/dialects/transform_extras.py +++ b/mlir/test/python/dialects/transform_extras.py @@ -30,7 +30,6 @@ def construct_and_print_in_module(f): with ir.InsertionPoint(module.body): f() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_gpu_ext.py b/mlir/test/python/dialects/transform_gpu_ext.py index 5771ebd13ab7..17f5a6d38f8c 100644 --- a/mlir/test/python/dialects/transform_gpu_ext.py +++ b/mlir/test/python/dialects/transform_gpu_ext.py @@ -19,7 +19,6 @@ def run(f): transform.YieldOp() print("\nTEST:", f.__name__) print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_loop_ext.py b/mlir/test/python/dialects/transform_loop_ext.py index 50c93fe9f46d..430b33fba04c 100644 --- a/mlir/test/python/dialects/transform_loop_ext.py +++ b/mlir/test/python/dialects/transform_loop_ext.py @@ -13,7 +13,6 @@ def run(f): print("\nTEST:", f.__name__) f() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_memref_ext.py b/mlir/test/python/dialects/transform_memref_ext.py index 55eceb57f97c..e7d871c9eac8 100644 --- a/mlir/test/python/dialects/transform_memref_ext.py +++ b/mlir/test/python/dialects/transform_memref_ext.py @@ -13,7 +13,6 @@ def run(f): print("\nTEST:", f.__name__) f() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_nvgpu_ext.py b/mlir/test/python/dialects/transform_nvgpu_ext.py index 45672ff3b58e..1de8b25bab7a 100644 --- a/mlir/test/python/dialects/transform_nvgpu_ext.py +++ b/mlir/test/python/dialects/transform_nvgpu_ext.py @@ -12,7 +12,6 @@ def run(f): print("\nTEST:", f.__name__) f() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_smt_ext.py b/mlir/test/python/dialects/transform_smt_ext.py index 0593421c7713..3692fd92344a 100644 --- a/mlir/test/python/dialects/transform_smt_ext.py +++ b/mlir/test/python/dialects/transform_smt_ext.py @@ -19,7 +19,6 @@ def run(f): f(sequence.bodyTarget) transform.YieldOp() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_sparse_tensor_ext.py b/mlir/test/python/dialects/transform_sparse_tensor_ext.py index b22f27ab6494..e11cc6bf1e07 100644 --- a/mlir/test/python/dialects/transform_sparse_tensor_ext.py +++ b/mlir/test/python/dialects/transform_sparse_tensor_ext.py @@ -19,7 +19,6 @@ def run(f): transform.YieldOp() print("\nTEST:", f.__name__) print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_structured_ext.py b/mlir/test/python/dialects/transform_structured_ext.py index ee15ac43e3d1..8785d6d36007 100644 --- a/mlir/test/python/dialects/transform_structured_ext.py +++ b/mlir/test/python/dialects/transform_structured_ext.py @@ -19,7 +19,6 @@ def run(f): f() module.operation.verify() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_tensor_ext.py b/mlir/test/python/dialects/transform_tensor_ext.py index 9dde72f52a8f..a2e7aa242b9d 100644 --- a/mlir/test/python/dialects/transform_tensor_ext.py +++ b/mlir/test/python/dialects/transform_tensor_ext.py @@ -19,7 +19,6 @@ def run(f): f(sequence.bodyTarget) transform.YieldOp() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_tune_ext.py b/mlir/test/python/dialects/transform_tune_ext.py index f8db151e1070..dfb93594bca5 100644 --- a/mlir/test/python/dialects/transform_tune_ext.py +++ b/mlir/test/python/dialects/transform_tune_ext.py @@ -19,7 +19,6 @@ def run(f): f(sequence.bodyTarget) transform.YieldOp() print(module) - module.operation.verify() return f diff --git a/mlir/test/python/dialects/transform_vector_ext.py b/mlir/test/python/dialects/transform_vector_ext.py index 0e841f2895b5..0cd9333dc121 100644 --- a/mlir/test/python/dialects/transform_vector_ext.py +++ b/mlir/test/python/dialects/transform_vector_ext.py @@ -21,7 +21,6 @@ def run_apply_patterns(f): transform.YieldOp() print("\nTEST:", f.__name__) print(module) - module.operation.verify() return f |