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

releaase 2.11 (#703)

Browse Source
This commit is contained in:
Aditya Atluri
2022-11-19 06:02:15 -08:00
committed by GitHub
parent 3c90f6aea6
commit c975e2ccbb
329 changed files with 47332 additions and 10607 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/library/include/cutlass/library/arch_mappings.h
View File

@ -97,6 +97,11 @@ template <typename OperatorClass> struct ArchMap<arch::Sm86, OperatorClass> {
static int const kMax = 1024;
};
template <typename OperatorClass> struct ArchMap<arch::Sm90, OperatorClass> {
static int const kMin = 90;
static int const kMax = 1024;
};
/////////////////////////////////////////////////////////////////////////////////////////////////
} // namespace library

3
tools/library/include/cutlass/library/library.h
View File

@ -1445,6 +1445,9 @@ struct ConvArguments {
/// pointer to implicit gemm matrix B
void const *B;
/// pointer to reordered matrix B
void const *reordered_B;
/// pointer to implicit gemm matrix C
void const *C;