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

Add conversion from ElementBias to ElementCompute (#961)

Browse Source
This commit is contained in:
Jack Kosaian
2023-05-26 23:08:36 -04:00
committed by GitHub
parent 6f47420213
commit 7dbf423763
4 changed files with 203 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/epilogue/thread/linear_combination_bias_elementwise.h
+1 -1
View File
@@ -95,7 +95,7 @@ public:
using FragmentSource = FragmentC;
using FragmentOutput = FragmentZ;
using ElementBias = ElementVector;
using FragmentBias = FragmentCompute;
using FragmentBias = Array<ElementBias, kElementsPerAccess>;
using ActivationFunctor = ElementwiseOp;
static const ScaleType::Kind kScale = ScaleType::Default;