Add set_k_partition function (#624)

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

View File

@ -90,6 +90,12 @@ public:
CUTLASS_HOST_DEVICE
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