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

Fixed template struct/class mismatch (#453)

Browse Source
This commit is contained in:
Stepan Tezyunichev 2022-04-24 23:30:21 +03:00 committed by GitHub
parent d7b499deff
commit be4578d517
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/cutlass/epilogue/warp/tile_iterator_volta_tensor_op.h
View File

@ -64,7 +64,7 @@ struct TileIteratorVoltaTensorOp;
template < template <
typename WarpShape_ ///< shape of warp-level GEMM (concept: MatrixShape) 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: public:
using WarpShape = WarpShape_; using WarpShape = WarpShape_;
@ -250,7 +250,7 @@ public:
template < template <
typename WarpShape_ ///< shape of warp-level GEMM (concept: MatrixShape) 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: public:
using WarpShape = WarpShape_; using WarpShape = WarpShape_;

2
include/cutlass/transform/thread/transpose.h
View File

@ -44,7 +44,7 @@ template <
int ElementCount, int ElementCount,
typename TransposeShape, typename TransposeShape,
typename Element typename Element
> class Transpose; > struct Transpose;
/// Specialization for int8_t 4x4 transpose /// Specialization for int8_t 4x4 transpose
template <int ElementCount_> template <int ElementCount_>