Function test_ex_8_7_deep_nested

tests/python/tirx/test_exec_context.py:250–263

()

def test_ex_8_7_deep_nested():
    ctx = ExecContext.at_kernel_entry(warp_ext=16).with_scope_switch(CTA)
    ctx = ctx.with_filter(WARP_FLAT, 0, 8).with_scope_switch(WARPGROUP)
    assert ctx.inter == {"wgid": AxisRange(2, 0), "cta_id": AxisRange(1, 0)}
    ctx = ctx.with_filter(WARP_FLAT, 0, 2)
    # recompute at warpgroup: factor(2, 0) -> case 2
    assert ctx.inter == {"wgid": AxisRange(1, 0), "cta_id": AxisRange(1, 0)}
    assert ctx.intra == {"laneid": AxisRange(32, 0), "wid_in_wg": AxisRange(2, 0)}
    ctx = ctx.with_scope_switch(WARP)
    assert ctx.inter == {"warpid": AxisRange(2, 0), "cta_id": AxisRange(1, 0)}
    assert ctx.intra == {"laneid": AxisRange(32, 0)}
    ctx = ctx.with_filter(LANE_BIND, 0, 8)
    assert ctx.intra == {"laneid": AxisRange(8, 0)}
    assert ctx.inter == {"warpid": AxisRange(2, 0), "cta_id": AxisRange(1, 0)}


# ---------------------------------------------------------------------------