CUTLASS 3.4.0 (#1286)
* CUTLASS 3.4.0 * Update CHANGELOG.md --------- Co-authored-by: Pradeep Ramani <prramani@nvidia.com>
This commit is contained in:
co-authored by
Pradeep Ramani
parent
b7508e3379
commit
8236f30675
@@ -67,9 +67,9 @@ class GemmUniversal<
|
||||
Epilogue_,
|
||||
ThreadblockSwizzle_,
|
||||
void,
|
||||
// 3.x kernels use the first template argument to define the ProblemShape tuple
|
||||
// 3.x kernels use the first template argument to define the ProblemShape
|
||||
// We use this invariant to SFINAE dispatch against either the 2.x API or the 3.x API
|
||||
cute::enable_if_t<not cute::is_tuple<Mma_>::value>
|
||||
cute::enable_if_t<not (cute::is_tuple<Mma_>::value || IsCutlass3ArrayKernel<Mma_>::value)>
|
||||
> {
|
||||
public:
|
||||
|
||||
|
||||
Reference in New Issue
Block a user