[bug] Cherry-pick "Fix dereferencing nullptr (taichi-dev#6763)" (taic…

…hi-dev#6769) Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
feisuzhu · Nov 30, 2022 · 0f25b95 · 0f25b95
1 parent e4a1129
commit 0f25b95
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/taichi/transforms/loop_invariant_detector.h b/taichi/transforms/loop_invariant_detector.h
@@ -65,6 +65,8 @@ class LoopInvariantDetector : public BasicStmtVisitor {
     bool is_invariant = true;
 
     for (Stmt *operand : stmt->get_operands()) {
+      if (operand == nullptr)
+        continue;
       is_invariant &= is_operand_loop_invariant_impl(operand, current_scope);
     }
 

diff --git a/taichi/transforms/make_thread_local.cpp b/taichi/transforms/make_thread_local.cpp
@@ -76,6 +76,8 @@ std::vector<std::pair<T *, AtomicOpType>> find_global_reduction_destinations(
             }
           }
           for (auto &op : stmt->get_operands()) {
+            if (op == nullptr)
+              continue;
             // Make sure the values of related atomic operations are not used.
             if (auto atomic = op->cast<AtomicOpStmt>()) {
               if (irpass::analysis::maybe_same_address(atomic->dest,