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

* [layout] Fix AffineRank2ColumnMajor::packed() * correct affine2row::packed --------- Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
zhangyuqin1998 · Mar 29, 2023 · bc36122 · bc36122
1 parent 15d9d31
commit bc36122
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.