Format

fc9b2a7d · Paul · d90d2137 · fc9b2a7d
Commit fc9b2a7d authored Sep 11, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

src/targets/gpu/jit/concat.cpp src/targets/gpu/jit/concat.cpp +4 -2

No files found.
--- a/src/targets/gpu/jit/concat.cpp
+++ b/src/targets/gpu/jit/concat.cpp
@@ -63,8 +63,10 @@ struct concat_compiler : compiler<concat_compiler>
    static std::size_t get_concat_elements(const std::vector<shape>& inputs)
    {
-        auto total = std::accumulate(
+        auto total =
-            inputs.begin(), std::prev(inputs.end()), 0, [](auto x, auto s) { return x + s.elements(); });
+            std::accumulate(inputs.begin(), std::prev(inputs.end()), 0, [](auto x, auto s) {
+                return x + s.elements();
+            });
        return total / (inputs.size() - 1);
    }