squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Merge pull request #264 from zheng95z/patch-3

Browse Source 
Adds `NoBetaScaling` for `LinearCombination`
This commit is contained in:
Haicheng Wu 2021-05-17 10:03:30 -04:00 committed by GitHub
commit b68113f5be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/cutlass/epilogue/thread/linear_combination.h
View File

@ -177,7 +177,11 @@ public:
multiplies<ComputeFragment> mul_add_source; multiplies<ComputeFragment> mul_add_source;
multiply_add<ComputeFragment> mul_add_accumulator; multiply_add<ComputeFragment> mul_add_accumulator;
if (Scale == ScaleType::NoBetaScaling)
intermediate = converted_source;
else
intermediate = mul_add_source(beta_, converted_source); // X = beta * C + uniform intermediate = mul_add_source(beta_, converted_source); // X = beta * C + uniform
intermediate = mul_add_accumulator(alpha_, converted_accumulator, intermediate); // D = alpha * Accum + X intermediate = mul_add_accumulator(alpha_, converted_accumulator, intermediate); // D = alpha * Accum + X
// Convert to destination numeric type // Convert to destination numeric type