NVIDIA/cutlass
1
0
Fork 0
mirror of https://github.com/NVIDIA/cutlass.git synced 2026-04-19 22:38:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix idx2crd docstring (#2914)

Browse Source 
* fix idx2crd docstring

* fix

* fix
This commit is contained in:
Wenxuan Tan
2026-01-07 10:11:38 -08:00
committed by GitHub
parent eb61c91147
commit f86feb0aa8
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/CuTeDSL/cutlass/cute/core.py
View File

@@ -2990,9 +2990,11 @@ def idx2crd(idx, shape, *, loc=None, ip=None):
@cute.jit
def foo():
coord = cute.idx2crd(11, (5,4))
# Computed as: 11 = 2 * 4 + 3, so coordinate is (2, 3)
# idx2crd is always col-major
# For shape (m, n, l, ...), coord = (idx % m, idx // m % n, idx // m // n % l, ...
# Computed as: (11 % 5, 11 // 5 % 4) = (1, 2)
print(coord)
foo() # Expected output: (2, 3)
foo() # Expected output: (1, 2)
**Note:**
Python DSL is aligned with C++ DSL.