Format

src/targets/gpu/jit/pointwise.cpp

@@ -56,13 +56,11 @@ struct pointwise_compiler : compiler<pointwise_compiler>
     static std::size_t compute_local(gen::vectorize v, const std::vector<shape>& inputs)
     {
         const std::size_t max_local = 1024;
-        if (std::none_of(inputs.begin(), inputs.end(), [&](auto s) {
-            return s.transposed();
-        }))
+        if(std::none_of(inputs.begin(), inputs.end(), [&](auto s) { return s.transposed(); }))
             return max_local;
-        if (std::any_of(inputs.begin(), inputs.end(), [&](auto s) {
-            return s.broadcasted() or s.strides()[v.axis] != 1;
-        }))
+        if(std::any_of(inputs.begin(), inputs.end(), [&](auto s) {
+               return s.broadcasted() or s.strides()[v.axis] != 1;
+           }))
             return max_local;
         return inputs.front().lens()[v.axis] / v.size;
     }
@@ -80,7 +78,8 @@ struct pointwise_compiler : compiler<pointwise_compiler>
             v,
             compute_global_for(ctx,
                                options.output.elements() / vec.size,
-                               oversubscribe_if(not preloads.is_preloading())), compute_local(vec, options.virtual_inputs));
+                               oversubscribe_if(not preloads.is_preloading())),
+            compute_local(vec, options.virtual_inputs));
         auto src = interpolate_string(pointwise_kernel,
                                       {{"params", enum_params(inputs.size(), "void * private_p")},
                                        {"args", enum_params(inputs.size(), "private_p")},