CUTLASS 2.6 (#298)

CUTLASS 2.6
2021-07-23 00:40:53 -04:00
parent 6c29fe20ba
commit e5d51840e8
308 changed files with 32408 additions and 4722 deletions
@@ -107,7 +107,8 @@ struct DefaultGemmWithReduction {
    EpilogueOutputOp,
    ThreadblockSwizzle,
    Stages,
-    Operator
+    Operator,
+    true 
  >::GemmKernel;

  // Replace epilogue
@@ -129,7 +130,6 @@ struct DefaultGemmWithReduction {
  >;
 };

-
 /////////////////////////////////////////////////////////////////////////////////////////////////

 /// Parital specialization: ArchTag = cutlass::arch::Sm70