CUTLASS 3.5.0 (#1411)

2024-03-19 17:51:04 -04:00
parent ffa34e7075
commit 629f4653c3
468 changed files with 48729 additions and 7252 deletions
@@ -30,9 +30,9 @@
 **************************************************************************************************/
 #pragma once

-#include <cute/config.hpp>
-#include <cute/util/type_traits.hpp>
 #include <cutlass/complex.h>
+#include <cute/util/type_traits.hpp>
+#include <cute/numeric/numeric_types.hpp>

 namespace cute
 {
@@ -44,6 +44,9 @@ using cutlass::real;
 using cutlass::imag;
 using cutlass::conj;

+template <class T>
+static constexpr auto is_complex_v = is_complex<T>::value;
+
 /// Fused multiply-add for complex numbers
 template <class T>
 CUTE_HOST_DEVICE constexpr