Address CR
This commit is contained in:
parent
83036ed646
commit
0b74c8f473
@ -134,6 +134,7 @@ public:
|
||||
/// Functionally required for serial reduction in the epilogue
|
||||
CUTLASS_HOST_DEVICE
|
||||
void set_k_partition(int k_partition, int k_partition_count) {
|
||||
CUTLASS_UNUSED(k_partition_count);
|
||||
if (k_partition) {
|
||||
beta_ = ElementCompute(1);
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user