gpu: jit: add missing check in reorder_nary_add_args()

oneapi-src · Apr 14, 2023 · 774deab · 774deab
1 parent 663c2e4
commit 774deab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gpu/jit/pass/simplify.cpp b/src/gpu/jit/pass/simplify.cpp
@@ -1499,7 +1499,8 @@ class common_factor_simplifier_t : public nary_op_mutator_t {
 
 expr_t reorder_nary_add_args(const expr_t &e, bool x64_first) {
     auto *nary_op = e.as_ptr<nary_op_t>();
-    if (nary_op->op_kind != op_kind_t::_add || nary_op->args.size() <= 2)
+    if (!nary_op || nary_op->op_kind != op_kind_t::_add
+            || nary_op->args.size() <= 2)
         return e;
 
     std::vector<expr_t> other_args;