MCPcopy Create free account
hub / github.com/apache/tvm / test_factor_case2_offset

Function test_factor_case2_offset

tests/python/tirx/test_exec_context.py:133–139  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

131
132
133def test_factor_case2_offset():
134 # warpid ext=2, off=6 -> wid_off=2, fits (2 <= 4-2)
135 A = initial_A(warp_ext=16)
136 A = filter_narrow(A, WARP_FLAT, 6, 8)
137 split = scope_switch(A, WARPGROUP)
138 assert split.inter["wgid"] == AxisRange(1, 1)
139 assert split.intra["wid_in_wg"] == AxisRange(2, 2)
140
141
142def test_factor_case3_fails():

Callers

nothing calls this directly

Calls 4

initial_AFunction · 0.90
filter_narrowFunction · 0.90
scope_switchFunction · 0.90
AxisRangeClass · 0.90

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…