Updates for CUTLASS 3.5.0 (#1468)

2024-04-11 21:33:40 -04:00
parent a40e08e9d5
commit 7d49e6c7e2
171 changed files with 7526 additions and 1888 deletions
--- a/python/cutlass_library/library.py
+++ b/python/cutlass_library/library.py
@@ -250,6 +250,12 @@ ComplexTransformTag = {
  ComplexTransform.conj: 'cutlass::ComplexTransform::kConjugate',
 }

+# Used for cutlass3x complex kernel collective mainloop builder instantiation
+ComplexTransformTag3x = {
+  ComplexTransform.none: 'cute::identity',
+  ComplexTransform.conj: 'cute::conjugate',
+}
+
 #
 RealComplexBijection = [
  (DataType.f16, DataType.cf16),