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

Add all supported GMMA shapes (#1890)

Browse Source
This commit is contained in:
Sergey Klevtsov 2024-10-22 15:13:36 -07:00 committed by GitHub
parent 5b50a8faaf
commit d65266a868
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
10 changed files with 158376 additions and 81391 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3670
include/cute/arch/mma_sm90.hpp
View File

File diff suppressed because it is too large Load Diff

29748
include/cute/arch/mma_sm90_gmma.hpp
View File

File diff suppressed because it is too large Load Diff

56445
include/cute/arch/mma_sm90_gmma_ext.hpp Normal file
View File

File diff suppressed because it is too large Load Diff

31698
include/cute/arch/mma_sm90_gmma_sparse.hpp
View File

File diff suppressed because it is too large Load Diff

60445
include/cute/arch/mma_sm90_gmma_sparse_ext.hpp Normal file
View File

File diff suppressed because it is too large Load Diff

11111
include/cute/atom/mma_traits_sm90_gmma.hpp
View File

File diff suppressed because it is too large Load Diff

20116
include/cute/atom/mma_traits_sm90_gmma_ext.hpp Normal file
View File

File diff suppressed because it is too large Load Diff

9185
include/cute/atom/mma_traits_sm90_gmma_sparse.hpp
View File

File diff suppressed because it is too large Load Diff

17335
include/cute/atom/mma_traits_sm90_gmma_sparse_ext.hpp Normal file
View File

File diff suppressed because it is too large Load Diff

14
include/cute/numeric/integral_constant.hpp
View File

@ -146,19 +146,33 @@ using _12 = Int<12>;
using _16 = Int<16>; using _16 = Int<16>;
using _24 = Int<24>; using _24 = Int<24>;
using _32 = Int<32>; using _32 = Int<32>;
using _40 = Int<40>;
using _48 = Int<48>; using _48 = Int<48>;
using _56 = Int<56>;
using _64 = Int<64>; using _64 = Int<64>;
using _72 = Int<72>;
using _80 = Int<80>; using _80 = Int<80>;
using _88 = Int<88>;
using _96 = Int<96>; using _96 = Int<96>;
using _104 = Int<104>;
using _112 = Int<112>; using _112 = Int<112>;
using _120 = Int<120>;
using _128 = Int<128>; using _128 = Int<128>;
using _136 = Int<136>;
using _144 = Int<144>; using _144 = Int<144>;
using _152 = Int<152>;
using _160 = Int<160>; using _160 = Int<160>;
using _168 = Int<168>;
using _176 = Int<176>; using _176 = Int<176>;
using _184 = Int<184>;
using _192 = Int<192>; using _192 = Int<192>;
using _200 = Int<200>;
using _208 = Int<208>; using _208 = Int<208>;
using _216 = Int<216>;
using _224 = Int<224>; using _224 = Int<224>;
using _232 = Int<232>;
using _240 = Int<240>; using _240 = Int<240>;
using _248 = Int<248>;
using _256 = Int<256>; using _256 = Int<256>;
using _384 = Int<384>; using _384 = Int<384>;
using _512 = Int<512>; using _512 = Int<512>;