Format

bf6cf5b0 · Paul · f7a59edb · bf6cf5b0
Commit bf6cf5b0 authored May 05, 2022 by Paul
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

src/targets/gpu/fuse_mlir.cpp src/targets/gpu/fuse_mlir.cpp +4 -3

No files found.
--- a/src/targets/gpu/fuse_mlir.cpp
+++ b/src/targets/gpu/fuse_mlir.cpp
@@ -29,8 +29,9 @@ struct find_conv_pointwise
        auto pm       = ins->module_inputs().front();
        auto names    = pm->get_parameter_names();
        // Whitelist pointwise operators
-        if (std::any_of(pm->begin(), pm->end(), [](const auto& i) {
+        if(std::any_of(pm->begin(), pm->end(), [](const auto& i) {
-            return not contains({"@literal", "@param", "@return", "convolution", "add", "relu"}, i.name());
+               return not contains({"@literal", "@param", "@return", "convolution", "add", "relu"},
+                                   i.name());
           }))
            return;
        std::sort(names.begin(), names.end());