Function test5

tests/python/s_tir/transform/test_s_tir_transform_profiling_instr.py:330–340

()

# test5: Use both 'lwp_min_height' and 'lwp_max_depth'.
# instrumentation.
def test5():
    test5_config = default_lwp_test_config.copy()
    test5_config.update(
        {"s_tir.lwp_max_depth": 3, "s_tir.instr_siblings": False, "s_tir.lwp_min_height": 2}
    )
    with tvm.transform.PassContext(config=test5_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"], test5_expected_output.with_attr("global_symbol", "main")
    )


# test6: Tests instrumentation for the parallel loops