Function test3

tests/python/s_tir/transform/test_s_tir_transform_profiling_instr.py:306–314

()

# loops are always excluded from instrumentation unless overwritten using
# 'lwp_min_height' (ex: test5)
def test3():
    test3_config = default_lwp_test_config.copy()
    test3_config.update({"s_tir.lwp_max_depth": 3, "s_tir.instr_siblings": False})
    with tvm.transform.PassContext(config=test3_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"], test3_expected_output.with_attr("global_symbol", "main")
    )


# test4: Use 'lwp_min_height' to exclude inner loops upto a certain height from