leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 26 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/gemm/kernel/default_gemm_with_broadcast.h
+2 -2
View File
@@ -121,7 +121,7 @@ struct DefaultGemmWithBroadcast {
GemmBase::Epilogue::kPartitionsK,
ElementC_,
typename EpilogueOutputOp::ElementT,
ElementC_,
typename EpilogueOutputOp::ElementVector,
EpilogueOutputOp,
GemmBase::Epilogue::kElementsPerAccess
>::Epilogue;
@@ -221,7 +221,7 @@ struct DefaultGemmWithBroadcast<
GemmBase::Epilogue::kPartitionsK,
ElementC_,
typename EpilogueOutputOp::ElementT,
ElementC_,
typename EpilogueOutputOp::ElementVector,
EpilogueOutputOp,
GemmBase::Epilogue::kElementsPerAccess
>::Epilogue;