Address CR

This commit is contained in:
KeDengMS 2021-04-19 23:36:06 +00:00
parent 83036ed646
commit 0b74c8f473

View File

@ -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);
}