CUTLASS 3.1 (#915)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
2023-04-14 23:19:34 -04:00
parent 9b8166e3f0
commit d572cc1aab
482 changed files with 37175 additions and 16410 deletions
@@ -66,7 +66,10 @@ void OperationTable::append(Manifest const &manifest) {
        gemm_desc.B.element,
        gemm_desc.B.layout,
        gemm_desc.transform_B,
-        gemm_desc.C.element
+        gemm_desc.C.element,
+        gemm_desc.C.layout,
+        gemm_desc.D.element,
+        gemm_desc.D.layout
      );

      Operation const *op = operation.get();