Fix clang tidy complexity issue

src/targets/gpu/device/include/migraph/gpu/device/nary.hpp

@@ -51,15 +51,9 @@ auto nary_nonstandard_impl(F f, argument result, Arguments... args)
     });
 }
 
-template <class... Arguments>
-auto nary_nonstandard(argument result, Arguments... args)
-{
-    return [=](auto f) { return nary_nonstandard_impl(f, result, args...); };
-}
-
-inline auto binary_broadcast_vec(const argument& result, const argument& arg1, const argument& arg2)
+template<class F>
+void binary_broadcast_vec_impl(F f, const argument& result, const argument& arg1, const argument& arg2)
 {
-    return [=](auto f) {
         const auto& output_shape = result.get_shape();
         const auto& b_shape      = arg2.get_shape();
         auto bdim                = std::distance(b_shape.strides().begin(),
@@ -106,12 +100,11 @@ inline auto binary_broadcast_vec(const argument& result, const argument& arg1, c
                 }
             });
         });
-    };
 }
 
-inline auto binary_broadcast(const argument& result, const argument& arg1, const argument& arg2)
+template<class F>
+void binary_broadcast_impl(F f, const argument& result, const argument& arg1, const argument& arg2)
 {
-    return [=](auto f) {
     const auto& output_shape = result.get_shape();
     const auto& b_shape      = arg2.get_shape();
     auto bdim                = std::distance(b_shape.strides().begin(),
@@ -150,13 +143,11 @@ inline auto binary_broadcast(const argument& result, const argument& arg1, const
             }
         });
     });
-    };
 }
 
-template <class... Arguments>
-auto nary_standard_vec(argument result, Arguments... args)
+template <class F, class... Arguments>
+void nary_standard_vec_impl(F f, argument result, Arguments... args)
 {
-    return [=](auto f) {
         // assert(x.get_shape().elements() == y.get_shape().elements());
         const auto& output_shape = result.get_shape();
         visit_all(result, args...)([&](auto output, auto... inputs) {
@@ -177,13 +168,11 @@ auto nary_standard_vec(argument result, Arguments... args)
                 outp[i] = out;
             });
         });
-    };
 }
 
-template <class... Arguments>
-auto nary_standard(argument result, Arguments... args)
+template <class F, class... Arguments>
+void nary_standard_impl(F f, argument result, Arguments... args)
 {
-    return [=](auto f) {
         // assert(x.get_shape().elements() == y.get_shape().elements());
         const auto& output_shape = result.get_shape();
         visit_all(result, args...)([&](auto output, auto... inputs) {
@@ -192,29 +181,44 @@ auto nary_standard(argument result, Arguments... args)
             gs_launch(output_shape.elements())(
                 [=](auto i) { data([&](auto... xps) { outp[i] = f(xps[i]...); }); });
         });
-    };
 }
 
-template <class... Arguments>
-auto nary_impl(argument result, Arguments... args)
+template <class F, class... Arguments>
+void nary_impl(F f, argument result, Arguments... args)
 {
-    return [=](auto f) {
         bool standard = all_of({args.get_shape()...}, [](const shape& s) { return s.standard(); });
         bool packed   = all_of({args.get_shape()...}, [](const shape& s) { return s.packed(); });
         bool same_shapes =
             all_of({args.get_shape()...}, [&](const shape& s) { return s == result.get_shape(); });
         if(standard or (packed and same_shapes))
-            nary_standard(result, args...)(f);
+            nary_standard_impl(f, result, args...);
         else
-            nary_nonstandard(result, args...)(f);
+            nary_nonstandard_impl(f, result, args...);
+
+}
+
+template <class... Arguments>
+auto nary_nonstandard(argument result, Arguments... args)
+{
+    return [=](auto f) { 
+        nary_nonstandard_impl(f, result, args...);
+    };
+}
 
+template <class... Arguments>
+auto nary_standard(argument result, Arguments... args)
+{
+    return [=](auto f) { 
+        nary_standard_impl(f, result, args...);
     };
 }
 
 template <class... Arguments>
 auto nary(argument result, Arguments... args)
 {
-    return nary_impl(result, args...);
+    return [=](auto f) { 
+        nary_impl(f, result, args...);
+    };
 }
 
 inline auto nary(const argument& result, const argument& arg1, const argument& arg2)
@@ -235,13 +239,13 @@ inline auto nary(const argument& result, const argument& arg1, const argument& a
                 const bool divisible_by_4 = (b_len % 4 == 0) and (b_stride % 4 == 0) and
                                             (arg1.get_shape().elements() % 4 == 0);
                 if(divisible_by_4)
-                    binary_broadcast_vec(result, arg1, arg2)(f);
+                    binary_broadcast_vec_impl(f, result, arg1, arg2);
                 else
-                    binary_broadcast(result, arg1, arg2)(f);
+                    binary_broadcast_impl(f, result, arg1, arg2);
                 return;
             }
         }
-        nary_impl(result, arg1, arg2)(f);
+        nary_impl(f, result, arg1, arg2);
     };
 }