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:
co-authored by
Haicheng Wu
parent
cc85b64cf6
commit
012c62c748
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user