[autodiff] Handle return statement (taichi-dev#7389)

Issue: taichi-dev#7378 ### Brief Summary
quadpixels · May 13, 2023 · e2b3f51 · e2b3f51
1 parent e04b866
commit e2b3f51
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
   }