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

Fix type bug in conv2d/gemm with broadcast (#796)

Browse Source 
add ElementVector

---------

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
Shuai Shao
2023-02-09 20:53:25 -05:00
committed by GitHub
co-authored by Haicheng Wu
parent 2e10404d26
commit ce8597dc14
7 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/conv/kernel/default_conv2d_fprop_with_broadcast.h
+1 -1
View File
@@ -107,7 +107,7 @@ struct DefaultConv2dFpropWithBroadcast {
ImplicitGemmBase::Epilogue::kPartitionsK,
ElementC,
typename EpilogueOutputOp::ElementT,
ElementC,
typename EpilogueOutputOp::ElementVector,
EpilogueOutputOp,
ImplicitGemmBase::Epilogue::kElementsPerAccess
>::Epilogue;