leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 29 Packages Projects Releases Wiki Activity

add support for sm89 in cute and the unit tests (#2177)

Browse Source 
* add support for sm89 in cute and the unit tests

* rebase v3.9 and format code

* minor fix

---------

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
kf-zhang
2025-04-10 14:16:36 -04:00
committed by GitHub
co-authored by Haicheng Wu
parent 09df6ac464
commit 19cc2a5feb
4 changed files with 349 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cute/atom/mma_atom.hpp
+1
View File
@@ -1111,6 +1111,7 @@ print_svg(TiledMMA<Args...> const &mma) {
#include <cute/atom/mma_traits_sm70.hpp>
#include <cute/atom/mma_traits_sm75.hpp>
#include <cute/atom/mma_traits_sm80.hpp>
#include <cute/atom/mma_traits_sm89.hpp>
#include <cute/atom/mma_traits_sm90.hpp>
#include <cute/atom/mma_traits_sm90_gmma.hpp>
#include <cute/atom/mma_traits_sm100.hpp>