Function test_scope_switch_warpgroup_aligned

tests/python/tirx/test_exec_context.py:103–109

()

def test_scope_switch_warpgroup_aligned():
    A = initial_A(warp_ext=16)
    split = scope_switch(A, WARPGROUP)
    assert split.inter["wgid"] == AxisRange(4, 0)
    assert split.inter["cta_id"] == AxisRange(1, 0)
    assert split.intra["laneid"] == AxisRange(32, 0)
    assert split.intra["wid_in_wg"] == AxisRange(4, 0)


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