Adds NoBetaScaling for LinearCombination

erees1 · May 12, 2021 · a68d7cd · a68d7cd
1 parent 38e8b29
commit a68d7cd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/include/cutlass/epilogue/thread/linear_combination.h b/include/cutlass/epilogue/thread/linear_combination.h
@@ -177,7 +177,11 @@ class LinearCombination {
     multiplies<ComputeFragment> mul_add_source;
     multiply_add<ComputeFragment> mul_add_accumulator;
 
-    intermediate = mul_add_source(beta_, converted_source);                             // X =  beta * C + uniform
+    if (Scale == ScaleType::NoBetaScaling)
+      intermediate = converted_source;
+    else
+      intermediate = mul_add_source(beta_, converted_source);                           // X =  beta * C + uniform
+
     intermediate = mul_add_accumulator(alpha_, converted_accumulator, intermediate);    // D = alpha * Accum + X
 
     // Convert to destination numeric type