opt env

PaddlePaddle · Jan 17, 2024 · 36a6b3b · 36a6b3b
1 parent 0ebfa25
commit 36a6b3b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/padiff/cinn_diff/env.py b/padiff/cinn_diff/env.py
@@ -89,16 +89,19 @@ def run_model(self, run_env, log):
 
     def run_base_model(self):
         self.init_base_env()
+        root_path = os.getcwd()
         os.chdir(self.script_path)
         run_env = self.base_env.copy()
         logger.info(run_env)
         run_env.update(self.os_env)
         base_log = open("base.log", "w")
         self.run_model(run_env, base_log)
         base_log.close()
+        os.chdir(root_path)
 
     def run_cinn_model(self):
         self.init_cinn_env()
+        root_path = os.getcwd()
         os.chdir(self.script_path)
         run_env = self.cinn_env.copy()
         base_env = self.base_env.copy()
@@ -110,3 +113,4 @@ def run_cinn_model(self):
         cinn_log = open("cinn.log", "w")
         self.run_model(run_env, cinn_log)
         cinn_log.close()
+        os.chdir(root_path)