[bug] Add GetElementExpression to offline cache key (taichi-dev#6918)

Issue: taichi-dev#4401, taichi-dev#6614 Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
PGZXB · Dec 19, 2022 · 1e6dd97 · 1e6dd97
1 parent d1a1728
commit 1e6dd97
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/taichi/analysis/gen_offline_cache_key.cpp b/taichi/analysis/gen_offline_cache_key.cpp
@@ -272,6 +272,12 @@ class ASTSerializer : public IRVisitor, public ExpressionVisitor {
     emit(expr->var);
   }
 
+  void visit(GetElementExpression *expr) override {
+    emit(ExprOpCode::GetElementExpression);
+    emit(expr->src);
+    emit(expr->index);
+  }
+
   void visit(Block *block) override {
     emit(StmtOpCode::EnterBlock);
     emit(static_cast<std::size_t>(block->statements.size()));

diff --git a/taichi/inc/expressions.inc.h b/taichi/inc/expressions.inc.h
@@ -24,3 +24,4 @@ PER_EXPRESSION(MeshIndexConversionExpression)
 PER_EXPRESSION(ReferenceExpression)
 PER_EXPRESSION(TextureOpExpression)
 PER_EXPRESSION(TexturePtrExpression)
+PER_EXPRESSION(GetElementExpression)