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

CUTLASS 3.0.0 (#786)

Browse Source 
* CUTLASS 3.0.0
This commit is contained in:
Vijay Thakkar
2023-01-23 20:55:28 -05:00
committed by GitHub
parent 66d9cddc83
commit 277bd6e537
377 changed files with 76396 additions and 1186 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/layout/matrix.h
+20
View File
@@ -39,6 +39,8 @@
*/
#pragma once
#include "cute/layout.hpp"
#include "cutlass/cutlass.h"
#include "cutlass/fast_math.h"
#include "cutlass/matrix_coord.h"
@@ -143,6 +145,15 @@ public:
LongIndex capacity(MatrixCoord const &extent) const {
return LongIndex(extent.row()) * LongIndex(stride_[0]);
}
CUTLASS_HOST_DEVICE
cute::Layout<cute::Shape<int, int>, cute::Stride<int64_t, cute::Int<1> > >
to_cute_layout(MatrixCoord const &extent) const {
return cute::Layout<cute::Shape<int, int>, cute::Stride<int64_t, cute::Int<1> > >{
{extent[0], extent[1]},
{stride(0), cute::Int<1>{}}
};
}
};
/// Mapping function for column-major matrices.
@@ -236,6 +247,15 @@ public:
LongIndex capacity(MatrixCoord const &extent) const {
return LongIndex(extent.column()) * LongIndex(stride_[0]);
}
CUTLASS_HOST_DEVICE
cute::Layout<cute::Shape<int, int>, cute::Stride< cute::Int<1>, int64_t> >
to_cute_layout(MatrixCoord const &extent) const {
return cute::Layout<cute::Shape<int, int>, cute::Stride<cute::Int<1>, int64_t> >{
{extent[0], extent[1]},
{cute::Int<1>{}, stride(0)}
};
}
};
/// Mapping function for interleaved matrices. Matrix is structured