[autodiff] Handle return statement (#7389)

Issue: #7378 ### Brief Summary
taichi-dev · Feb 23, 2023 · 6170861 · 6170861
1 parent 499834b
commit 6170861
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/taichi/transforms/auto_diff.cpp b/taichi/transforms/auto_diff.cpp
@@ -635,6 +635,10 @@ class ADTransform : public IRVisitor {
     // do nothing
   }
 
+  void visit(ReturnStmt *stmt) override {
+    // do nothing
+  }
+
   void visit(WhileControlStmt *stmt) override {
     TI_NOT_IMPLEMENTED
   }