[layout] Fix AffineRank2ColumnMajor::packed() (#879)

* [layout] Fix AffineRank2ColumnMajor::packed() * correct affine2row::packed --------- Co-authored-by: Haicheng Wu <[email protected]>
SNU-ARC · Feb 7, 2024 · 8877318 · 8877318
1 parent b2b17db
commit 8877318
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/include/cutlass/layout/matrix.h b/include/cutlass/layout/matrix.h
@@ -805,7 +805,7 @@ struct AffineRank2ColumnMajor {
   /// Helper returns a layout to a tightly packed tensor
   CUTLASS_HOST_DEVICE
   static AffineRank2ColumnMajor packed(MatrixCoord const &extent) {
-    return AffineRank2ColumnMajor(extent.column(), 1);
+    return AffineRank2ColumnMajor(1, extent.row());
   }
 
   /// Returns the offset of a coordinate in linear memory. 
@@ -911,7 +911,7 @@ struct AffineRank2RowMajor {
   /// Helper returns a layout to a tightly packed tensor
   CUTLASS_HOST_DEVICE
   static AffineRank2RowMajor packed(MatrixCoord const &extent) {
-    return AffineRank2RowMajor(extent.column(), 1);
+    return AffineRank2RowMajor(1, extent.row());
   }
 
   /// Returns the offset of a coordinate in linear memory.