From a68d7cd6f14b1867dbb5733f3802f604f732974e Mon Sep 17 00:00:00 2001 From: Zheng Zeng Date: Wed, 12 May 2021 22:23:55 +0800 Subject: [PATCH] Adds `NoBetaScaling` for `LinearCombination` --- include/cutlass/epilogue/thread/linear_combination.h | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/include/cutlass/epilogue/thread/linear_combination.h b/include/cutlass/epilogue/thread/linear_combination.h index fa2f72ac..0be9fc47 100644 --- a/include/cutlass/epilogue/thread/linear_combination.h +++ b/include/cutlass/epilogue/thread/linear_combination.h @@ -177,7 +177,11 @@ public: multiplies mul_add_source; multiply_add 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