41 lines · python
1# RUN: %PYTHON %s | FileCheck %s2 3from mlir.ir import *4from mlir.dialects import transform5from mlir.dialects.transform import tensor6 7 8def run(f):9 print("\nTEST:", f.__name__)10 with Context(), Location.unknown():11 module = Module.create()12 with InsertionPoint(module.body):13 sequence = transform.SequenceOp(14 transform.FailurePropagationMode.Propagate,15 [],16 transform.AnyOpType.get(),17 )18 with InsertionPoint(sequence.body):19 f(sequence.bodyTarget)20 transform.YieldOp()21 print(module)22 return f23 24 25@run26def testMakeLoopIndependentOpCompact(target):27 tensor.MakeLoopIndependentOp(target, 4)28 # CHECK-LABEL: TEST: testMakeLoopIndependentOpCompact29 # CHECK: = transform.tensor.make_loop_independent30 # CHECK-SAME: num_loops = 4 : i6431 # CHECK-SAME: (!transform.any_op) -> !transform.any_op32 33 34@run35def testMakeLoopIndependentOpTyped(target):36 tensor.MakeLoopIndependentOp(transform.OperationType.get("test.dummy"), target, 4)37 # CHECK-LABEL: TEST: testMakeLoopIndependentOpTyped38 # CHECK: = transform.tensor.make_loop_independent39 # CHECK-SAME: num_loops = 4 : i6440 # CHECK-SAME: (!transform.any_op) -> !transform.op<"test.dummy">41