Skip to content

Commit

Permalink
Add all supported GMMA shapes (#1890)
Browse files Browse the repository at this point in the history
  • Loading branch information
sklevtsov-nvidia authored Oct 22, 2024
1 parent 5b50a8f commit d65266a
Show file tree
Hide file tree
Showing 10 changed files with 180,598 additions and 103,613 deletions.
3,678 changes: 3,674 additions & 4 deletions include/cute/arch/mma_sm90.hpp

Large diffs are not rendered by default.

47,002 changes: 8,631 additions & 38,371 deletions include/cute/arch/mma_sm90_gmma.hpp

Large diffs are not rendered by default.

56,445 changes: 56,445 additions & 0 deletions include/cute/arch/mma_sm90_gmma_ext.hpp

Large diffs are not rendered by default.

49,616 changes: 9,285 additions & 40,331 deletions include/cute/arch/mma_sm90_gmma_sparse.hpp

Large diffs are not rendered by default.

60,445 changes: 60,445 additions & 0 deletions include/cute/arch/mma_sm90_gmma_sparse_ext.hpp

Large diffs are not rendered by default.

16,153 changes: 2,538 additions & 13,615 deletions include/cute/atom/mma_traits_sm90_gmma.hpp

Large diffs are not rendered by default.

20,116 changes: 20,116 additions & 0 deletions include/cute/atom/mma_traits_sm90_gmma_ext.hpp

Large diffs are not rendered by default.

13,407 changes: 2,115 additions & 11,292 deletions include/cute/atom/mma_traits_sm90_gmma_sparse.hpp

Large diffs are not rendered by default.

17,335 changes: 17,335 additions & 0 deletions include/cute/atom/mma_traits_sm90_gmma_sparse_ext.hpp

Large diffs are not rendered by default.

14 changes: 14 additions & 0 deletions include/cute/numeric/integral_constant.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -146,19 +146,33 @@ using _12 = Int<12>;
using _16 = Int<16>;
using _24 = Int<24>;
using _32 = Int<32>;
using _40 = Int<40>;
using _48 = Int<48>;
using _56 = Int<56>;
using _64 = Int<64>;
using _72 = Int<72>;
using _80 = Int<80>;
using _88 = Int<88>;
using _96 = Int<96>;
using _104 = Int<104>;
using _112 = Int<112>;
using _120 = Int<120>;
using _128 = Int<128>;
using _136 = Int<136>;
using _144 = Int<144>;
using _152 = Int<152>;
using _160 = Int<160>;
using _168 = Int<168>;
using _176 = Int<176>;
using _184 = Int<184>;
using _192 = Int<192>;
using _200 = Int<200>;
using _208 = Int<208>;
using _216 = Int<216>;
using _224 = Int<224>;
using _232 = Int<232>;
using _240 = Int<240>;
using _248 = Int<248>;
using _256 = Int<256>;
using _384 = Int<384>;
using _512 = Int<512>;
Expand Down

0 comments on commit d65266a

Please sign in to comment.