Improve sm90 mixed dtype kernel (#1883)

2024-10-17 17:06:38 -07:00
parent 755194a7bd
commit 08101d9d0c
11 changed files with 994 additions and 80 deletions
--- a/include/cute/algorithm/tuple_algorithms.hpp
+++ b/include/cute/algorithm/tuple_algorithms.hpp
@ -340,7 +340,7 @@ auto
 all_of(T const& t, F&& f)
 {
  if constexpr (is_tuple<T>::value) {
-    return detail::apply(t, [&] (auto const&... a) { return (true_type{} && ... && f(a)); }, tuple_seq<T>{});
+    return detail::apply(cute::transform(t, f), [&] (auto const&... a) { return (true_type{} && ... && a); }, tuple_seq<T>{});
  } else {
    return f(t);
  }