gpu: jit: conv: fix shuffle propagation

oneapi-src · Oct 31, 2022 · 90be8d5 · 90be8d5
1 parent caf4863
commit 90be8d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/gpu/jit/conv/ir_simplify.cpp b/src/gpu/jit/conv/ir_simplify.cpp
@@ -2073,14 +2073,15 @@ expr_t simplify_propagate_shuffle(const expr_t &e) {
         bool found_binary = false;
         for (int i : shuffle->idx) {
             if (is_binary_op(shuffle->vec[i])) {
-                found_binary = true;
                 auto &op = shuffle->vec[i].as<binary_op_t>();
+                if (found_binary && op.op_kind != op_kind_t::_and) continue;
+                found_binary = true;
                 a_type = op.a.type();
                 b_type = op.b.type();
                 op_kind = op.op_kind;
                 if (is_const(op.a)) a_common_const = op.a;
                 if (is_const(op.b)) b_common_const = op.b;
-                break;
+                if (op_kind == op_kind_t::_and) break;
             }
         }
         if (!found_binary) return e;