CUTLASS 3.5.1 (#1623)

* CUTLASS 3.5.1

* updates, optimizations, fixes

include/cute/numeric/complex.hpp

@@ -48,13 +48,13 @@ template <class T>
 static constexpr auto is_complex_v = is_complex<T>::value;
 
 /// Fused multiply-add for complex numbers
-template <class T>
+template <class D, class A, class B, class C>
 CUTE_HOST_DEVICE constexpr
 void
-fma(complex<T>      & d,
-    complex<T> const& a,
-    complex<T> const& b,
-    complex<T> const& c)
+fma(complex<D>      & d,
+    complex<A> const& a,
+    complex<B> const& b,
+    complex<C> const& c)
 {
   fma(d.real(),  a.real(), b.real(), c.real());
   fma(d.imag(),  a.real(), b.imag(), c.imag());
@@ -63,12 +63,12 @@ fma(complex<T>      & d,
 }
 
 /// Fused multiply-add for triplets
-template <class T>
+template <class A, class B, class C>
 CUTE_HOST_DEVICE constexpr
 void
-fma(complex<T> const& a,
-    complex<T> const& b,
-    complex<T>      & c)
+fma(complex<A> const& a,
+    complex<B> const& b,
+    complex<C>      & c)
 {
   return fma(c, a, b, c);
 }