MCPcopy Index your code
hub / github.com/apache/tvm / case_compose_slice

Function case_compose_slice

tests/python/tirx/test_layout.py:1629–1639  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

1627 case_swizzle_slice()
1628
1629 def case_compose_slice():
1630 # ComposeLayout slice
1631 compose = ComposeLayout(
1632 SwizzleLayout(per_element=3, swizzle_len=3, atom_len=3),
1633 TileLayout(S[(8, 64) : (64, 1)]),
1634 )
1635 shape = [512]
1636 region = [(64, 128)]
1637 sliced = compose.slice(shape, region)
1638 assert sliced is not None
1639 verify_slice(compose, shape, region, sliced)
1640
1641 case_compose_slice()
1642

Callers 1

test_sliceFunction · 0.85

Calls 5

ComposeLayoutClass · 0.90
SwizzleLayoutClass · 0.90
TileLayoutClass · 0.90
verify_sliceFunction · 0.70
sliceMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…