Function test_scope_switch_trivial

tests/python/tirx/test_exec_context.py:96–100

(kappa, expected_inter_axes, expected_intra_axes)

    ],
)
def test_scope_switch_trivial(kappa, expected_inter_axes, expected_intra_axes):
    A = initial_A(warp_ext=16, cta_ext=4)
    split = scope_switch(A, kappa)
    assert set(split.inter) == expected_inter_axes
    assert set(split.intra) == expected_intra_axes


def test_scope_switch_warpgroup_aligned():