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-13 22:34:20 -04:00
committed by GitHub
parent 50ceed7154
commit 2cc2c7ba1f
@@ -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