clean up multiple ConstStmt replacement

taichi-dev · Aug 16, 2021 · 4994e01 · 4994e01
1 parent 87667a7
commit 4994e01
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/taichi/transforms/offload.cpp b/taichi/transforms/offload.cpp
@@ -520,18 +520,13 @@ class FixCrossOffloadReferences : public BasicStmtVisitor {
       return false;
     if (stmt_to_offloaded[stmt] == stmt_to_offloaded[op])  // same OffloadedStmt
       return false;
-    if (op->is<ConstStmt>()) {
-      auto copy = op->as<ConstStmt>()->copy();
-      stmt_to_offloaded[copy.get()] = stmt_to_offloaded[stmt];
-      stmt->set_operand(index, copy.get());
-      stmt->insert_before_me(std::move(copy));
-    }
     if (op->is<GlobalPtrStmt>()) {
       auto copy = op->clone();
       copy->as<GlobalPtrStmt>()->activate = false;
       stmt_to_offloaded[copy.get()] = stmt_to_offloaded[stmt];
       stmt->set_operand(index, copy.get());
       stmt->insert_before_me(std::move(copy));
+      return true;
     }
 
     if (local_to_global_offset.find(op) == local_to_global_offset.end()) {