Function test2

tests/python/s_tir/transform/test_s_tir_transform_profiling_instr.py:291–299

()

# By default, only loops with siblings are instrumented. Here, 'lwp_max_depth'
# doesn't have any effect unless 'instr_siblings' is set to False (ex: test3).
def test2():
    test2_config = default_lwp_test_config.copy()
    test2_config.update({"s_tir.lwp_max_depth": 3})
    with tvm.transform.PassContext(config=test2_config):
        mod = tvm.IRModule.from_expr(input1.with_attr("global_symbol", "main"))
        mod = tvm.s_tir.transform.InstrumentProfileIntrinsics()(mod)
    tvm.ir.assert_structural_equal(
        mod["main"], test1_expected_output.with_attr("global_symbol", "main")
    )


# test3: Use 'lwp_max_depth' to instrument loops upto a certain depth. This flag