bug fixes and enharcement to gemm reductionK fusion (#682)

* add two missing files

* fix bunch of bugs of gemm-reducek fusion and add a device interface

* small changes

Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
Haicheng Wu
2022-11-03 11:07:50 -04:00
committed by GitHub
co-authored by Haicheng Wu
parent cc85b64cf6
commit 012c62c748
8 changed files with 445 additions and 21 deletions
@@ -91,7 +91,7 @@ template <
typename ElementAccumulator,
/// Operator class tag
typename OperatorClass,
///
/// Reduce A or B along the K dimension
bool ReduceKForA_,
/// Tag indicating architecture to tune for
typename ArchTag,