leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 26 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
tools/library/include/cutlass/library/library.h
+16 -2
View File
@@ -354,6 +354,9 @@ struct TileDescription {
/// Minimum compute capability (e.g. 70, 75) of a device eligible to run the operation.
int maximum_compute_capability;
/// Describes the shape of a cluster (in blocks)
cutlass::gemm::GemmCoord cluster_shape;
//
// Methods
//
@@ -364,14 +367,16 @@ struct TileDescription {
cutlass::gemm::GemmCoord warp_count = cutlass::gemm::GemmCoord(),
MathInstructionDescription math_instruction = MathInstructionDescription(),
int minimum_compute_capability = 0,
int maximum_compute_capability = 0
int maximum_compute_capability = 0,
cutlass::gemm::GemmCoord cluster_shape = cutlass::gemm::GemmCoord(1,1,1)
):
threadblock_shape(threadblock_shape),
threadblock_stages(threadblock_stages),
warp_count(warp_count),
math_instruction(math_instruction),
minimum_compute_capability(minimum_compute_capability),
maximum_compute_capability(maximum_compute_capability) { }
maximum_compute_capability(maximum_compute_capability),
cluster_shape(cluster_shape) { }
// Equality operator
inline
@@ -991,6 +996,9 @@ struct GemmUniversalConfiguration {
};
struct GemmUniversalArguments {
// NOTE: these are replicated for 3.0 interfaces
gemm::GemmCoord problem_size;
int batch_count;
void const *A;
void const *B;
@@ -1001,6 +1009,12 @@ struct GemmUniversalArguments {
void const *beta;
ScalarPointerMode pointer_mode;
// NOTE: these are replicated for 3.0 interfaces
int64_t lda;
int64_t ldb;
int64_t ldc;
int64_t ldd;
int64_t batch_stride_A;
int64_t batch_stride_B;
int64_t batch_stride_C;