This is an automated email from the ASF dual-hosted git repository.
spectrometerHBH pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 115029b3ab [FIX][TIRx] Constant-fold copy slice extents (#20067)
115029b3ab is described below
commit 115029b3ab44c177bfca56c2c1bcf65566450d26
Author: Hongyi Jin <[email protected]>
AuthorDate: Wed Jul 29 14:06:18 2026 -0400
[FIX][TIRx] Constant-fold copy slice extents (#20067)
## Motivation and context
Copy regions can retain constant-valued expressions after variable
substitution. Converting those expressions directly to Python integers
rejects valid slices even though their extents are statically known.
## Changes
- Simplify swizzled-region extents before converting them to Python
integers.
- Continue rejecting genuinely symbolic extents.
- Add a hardware-independent regression to the existing layout test
suite.
## Testing
- `python -m pytest tests/python/tirx/test_layout.py`
- Changed-files pre-commit checks
---
.../tvm/backend/cuda/operator/tile_primitive/copy/_common.py | 6 +++++-
tests/python/tirx/test_layout.py | 11 +++++++++++
2 files changed, 16 insertions(+), 1 deletion(-)
diff --git a/python/tvm/backend/cuda/operator/tile_primitive/copy/_common.py
b/python/tvm/backend/cuda/operator/tile_primitive/copy/_common.py
index b1dd018f86..5ff8100c39 100644
--- a/python/tvm/backend/cuda/operator/tile_primitive/copy/_common.py
+++ b/python/tvm/backend/cuda/operator/tile_primitive/copy/_common.py
@@ -320,7 +320,11 @@ def _extract_tile(layout, region):
if isinstance(layout, ComposeLayout):
return layout.tile_layout
if isinstance(layout, SwizzleLayout):
- extents = [int(end - start) for (start, end) in region]
+ # Region bounds may be constant-valued but remain as unfolded
+ # expressions after substitution. Simplify before converting to a
+ # Python integer; genuinely symbolic tile extents still raise.
+ analyzer = arith.Analyzer()
+ extents = [int(analyzer.simplify(end - start)) for (start, end) in
region]
return TileLayout(S[tuple(extents)])
return layout
diff --git a/tests/python/tirx/test_layout.py b/tests/python/tirx/test_layout.py
index 5099a5e324..998eab97d0 100644
--- a/tests/python/tirx/test_layout.py
+++ b/tests/python/tirx/test_layout.py
@@ -1655,6 +1655,17 @@ def test_slice():
case_compose_slice_2d()
+def test_cuda_copy_extract_swizzle_tile_simplifies_constant_region_extents():
+ from tvm.tirx.cuda.operator.tile_primitive.copy._common import
_extract_tile
+
+ zero = tvm.tirx.Mul(tvm.tirx.IntImm("int32", 0), tvm.tirx.IntImm("int32",
64))
+ end = tvm.tirx.Add(zero, tvm.tirx.IntImm("int32", 64))
+
+ tile = _extract_tile(SwizzleLayout(3, 3, 3), [(zero, end)])
+
+ assert [int(it.extent) for it in tile.shard] == [64]
+
+
def test_apply_to_shape():
"""``apply_to_shape`` should give per-shard coord, preferring per-dim
split when the input shape aligns with the layout's grouping."""
