Fixed template struct/class mismatch (NVIDIA#453)

lessw2020 · Apr 24, 2022 · be4578d · be4578d
1 parent d7b499d
commit be4578d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/include/cutlass/epilogue/warp/tile_iterator_volta_tensor_op.h b/include/cutlass/epilogue/warp/tile_iterator_volta_tensor_op.h
@@ -64,7 +64,7 @@ struct TileIteratorVoltaTensorOp;
 template <
   typename WarpShape_         ///< shape of warp-level GEMM (concept: MatrixShape)
 >
-class TileIteratorVoltaTensorOp<WarpShape_, gemm::GemmShape<32, 32, 4>, half_t, layout::RowMajor> {
+struct TileIteratorVoltaTensorOp<WarpShape_, gemm::GemmShape<32, 32, 4>, half_t, layout::RowMajor> {
 public:
 
   using WarpShape = WarpShape_;
@@ -250,7 +250,7 @@ class TileIteratorVoltaTensorOp<WarpShape_, gemm::GemmShape<32, 32, 4>, half_t,
 template <
   typename WarpShape_         ///< shape of warp-level GEMM (concept: MatrixShape)
 >
-class TileIteratorVoltaTensorOp<WarpShape_, gemm::GemmShape<32, 32, 4>, float, layout::RowMajor> {
+struct TileIteratorVoltaTensorOp<WarpShape_, gemm::GemmShape<32, 32, 4>, float, layout::RowMajor> {
 public:
 
   using WarpShape = WarpShape_;

diff --git a/include/cutlass/transform/thread/transpose.h b/include/cutlass/transform/thread/transpose.h
@@ -44,7 +44,7 @@ template <
   int ElementCount, 
   typename TransposeShape, 
   typename Element
-> class Transpose;
+> struct Transpose;
 
 /// Specialization for int8_t 4x4 transpose
 template <int ElementCount_>