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

Update mma_atom.hpp (#2159)

Browse Source 
remove useless code
This commit is contained in:
liujshi
2025-04-03 11:42:10 -04:00
committed by GitHub
parent 79fc51f4b8
commit df8a550d39
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cute/atom/mma_atom.hpp
-4
View File
@@ -450,8 +450,6 @@ struct TiledMMA : MMA_Atom
{ {
// (M,K) -> (M,K) // (M,K) -> (M,K)
auto ref_A = make_layout(make_shape(tile_size_mnk<0>(), tile_size_mnk<2>())); auto ref_A = make_layout(make_shape(tile_size_mnk<0>(), tile_size_mnk<2>()));
// (athrid,val) -> (M,K)
auto layoutA_TV = thrfrg_A(ref_A);
// (ThrV,(ThrM,ThrK)) -> (ThrV,(ThrM,ThrN,ThrK)) // (ThrV,(ThrM,ThrK)) -> (ThrV,(ThrM,ThrN,ThrK))
auto atile = make_tile(_, auto atile = make_tile(_,
@@ -489,8 +487,6 @@ struct TiledMMA : MMA_Atom
{ {
// (N,K) -> (N,K) // (N,K) -> (N,K)
auto ref_B = make_layout(make_shape(tile_size_mnk<1>(), tile_size_mnk<2>())); auto ref_B = make_layout(make_shape(tile_size_mnk<1>(), tile_size_mnk<2>()));
// (bthrid,val) -> (N,K)
auto layoutB_TV = thrfrg_B(ref_B);
// (ThrV,(ThrN,ThrK)) -> (ThrV,(ThrM,ThrN,ThrK)) // (ThrV,(ThrN,ThrK)) -> (ThrV,(ThrM,ThrN,ThrK))
auto btile = make_tile(_, auto btile = make_tile(_,