Function test_ex_8_5_partial_wg

tests/python/tirx/test_exec_context.py:222–227

()

def test_ex_8_5_partial_wg():
    ctx = ExecContext.at_kernel_entry(warp_ext=16).with_scope_switch(CTA)
    ctx = ctx.with_filter(WARP_FLAT, 0, 2).with_scope_switch(WARPGROUP)
    # case 2: 2 <= 4-0
    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)}


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