46 lines · python
1# RUN: %PYTHON %s | FileCheck %s2 3from mlir.ir import *4from mlir.dialects import transform5from mlir.dialects.transform import debug6 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 testDebugEmitParamAsRemark(target):27 i0 = IntegerAttr.get(IntegerType.get_signless(32), 0)28 i0_param = transform.ParamConstantOp(transform.AnyParamType.get(), i0)29 debug.emit_param_as_remark(i0_param)30 debug.emit_param_as_remark(i0_param, anchor=target, message="some text")31 # CHECK-LABEL: TEST: testDebugEmitParamAsRemark32 # CHECK: ^{{.*}}(%[[ARG0:.+]]: !transform.any_op):33 # CHECK: %[[PARAM:.*]] = transform.param.constant34 # CHECK: transform.debug.emit_param_as_remark %[[PARAM]]35 # CHECK: transform.debug.emit_param_as_remark %[[PARAM]]36 # CHECK-SAME: "some text"37 # CHECK-SAME: at %[[ARG0]]38 39 40@run41def testDebugEmitRemarkAtOp(target):42 debug.emit_remark_at(target, "some text")43 # CHECK-LABEL: TEST: testDebugEmitRemarkAtOp44 # CHECK: ^{{.*}}(%[[ARG0:.+]]: !transform.any_op):45 # CHECK: transform.debug.emit_remark_at %[[ARG0]], "some text"46