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

Add set_k_partition function (#624)

Browse Source 
A member function set_k_partition is required for the instatiation of cutlass::gemm::kernel::Gemm, even though SplitKSerial is false
This commit is contained in:
alexfreudenberg 2022-09-14 04:34:20 +02:00 committed by GitHub
parent 50ceed7154
commit 2cc2c7ba1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
include/cutlass/epilogue/thread/conversion_op.h
View File

@ -90,6 +90,12 @@ public:
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
Convert(Params const &params = Params()) { Convert(Params const &params = Params()) {
}
/// Functionally required for serial reduction in the epilogue
CUTLASS_HOST_DEVICE
void set_k_partition(int k_partition, int k_partition_count) {
} }
/// Returns true if source is needed based on state of runtime arguments /// Returns true if source is needed based on state of runtime arguments