Format

84327e69 · Paul · 86b49567 · 84327e69
Commit 84327e69 authored Jun 02, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

src/targets/gpu/compile_gen.cpp src/targets/gpu/compile_gen.cpp +2 -3

No files found.
--- a/src/targets/gpu/compile_gen.cpp
+++ b/src/targets/gpu/compile_gen.cpp
@@ -20,9 +20,8 @@ static std::vector<std::size_t> vector_sizes(const std::vector<shape>& inputs)
 vectorize vectorize::elements(std::size_t axis, const std::vector<shape>& inputs)
 {
-    if (std::all_of(inputs.begin(), inputs.end(), [&](const auto& s) {
+    if(std::all_of(
-            return s.lens()[axis] == 1;
+           inputs.begin(), inputs.end(), [&](const auto& s) { return s.lens()[axis] == 1; }))
-        }))
        return {1, axis};
    auto sizes = vector_sizes(inputs);
    std::vector<std::size_t> max_vec_size;