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:
parent
50ceed7154
commit
2cc2c7ba1f
@ -90,6 +90,12 @@ public:
|
||||
CUTLASS_HOST_DEVICE
|
||||
Convert(Params const ¶ms = 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
|
||||
|
Loading…
Reference in New Issue
Block a user