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

v3.9 (#2185)

Browse Source 
* v3.8 update x

* fix blackwell gg

* doc change

* doc change

* doc change

---------

Co-authored-by: yuzhai <yuzhai@nvidia.com>
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
Co-authored-by: Haicheng Wu <57973641+hwu36@users.noreply.github.com>
This commit is contained in:
Yujia Zhai
2025-03-20 22:52:23 -07:00
committed by GitHub
parent 8c4d1dc47d
commit 62750a2b75
334 changed files with 91517 additions and 2656 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/python/pycute/test_composition.py
View File

@@ -199,6 +199,15 @@ class TestComposition(unittest.TestCase):
layoutB = Layout((4,2,2), (2,8,1))
self.helper_test_composition(layoutA, layoutB)
# Pre-coalesced LHS
layoutA = Layout((4,6,8),(1,4,7))
layoutB = Layout((6),(1))
self.helper_test_composition(layoutA, layoutB)
# Mid-layout truncation
layoutA = Layout((4,6,8,10),(2,3,5,7))
layoutB = Layout(6,12)
self.helper_test_composition(layoutA, layoutB)
if __name__ == "__main__":
unittest.main()