plt update net instance, test=model

PaddlePaddle · Sep 3, 2024 · e71139f · e71139f
1 parent ce0c1e7
commit e71139f
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/framework/e2e/PaddleLT_new/binary_search.py b/framework/e2e/PaddleLT_new/binary_search.py
@@ -62,6 +62,7 @@ def __init__(self, commit_list, title, layerfile, testing, perf_decay=None, test
         self.test_obj = test_obj
         self.py_cmd = os.environ.get("python_ver")
         self.testing_mode = os.environ.get("TESTING_MODE")
+        self.device_place_id = 0
 
     def _status_print(self, exit_code, status_str):
         """
@@ -82,8 +83,8 @@ def _install_paddle(self, commit_id):
 
         whl_link = (
             "https://paddle-qa.bj.bcebos.com/paddle-pipe"
-            "line/Develop-GpuAll-LinuxCentos-Gcc82-Cuda112-Trtoff-Py38-Compile/{}/paddle"
-            "paddle_gpu-0.0.0-cp38-cp38-linux_x86_64.whl".format(commit_id)
+            "line/Develop-GpuSome-LinuxCentos-Gcc82-Cuda118-Cudnn86-Trt85-Py310-CINN-Compile/{}/paddle"
+            "paddle_gpu-0.0.0-cp310-cp310-linux_x86_64.whl".format(commit_id)
         )
         exit_code = os.system(f"{self.py_cmd} -m pip install {whl_link}")
         self._status_print(exit_code=exit_code, status_str="install paddlepaddle-gpu")
@@ -105,7 +106,7 @@ def _precision_debug(self, commit_id):
         exit_code = os.system(
             f"cp -r PaddleLT.py {self.title}.py && "
             f"{self.py_cmd} -m pytest {self.title}.py --title={self.title} "
-            f"--layerfile={self.layerfile} --testing={self.testing}"
+            f"--layerfile={self.layerfile} --testing={self.testing} --device_place_id={self.device_place_id}"
         )
 
         if exit_code > 0:
@@ -168,8 +169,8 @@ def _commit_locate(self, commits):
     if not os.path.exists("paddle"):
         os.system("git clone -b develop http://github.com/paddlepaddle/paddle.git")
     os.chdir(os.path.join(cur_path, "paddle"))
-    start_commit = "f651ac5cf387c56488b52fcc6456a63e9eb73086"  # 成功commit
-    end_commit = "9d5f31687cce16a976256723a70df4550085d685"  # 失败commit
+    start_commit = "4ac66e4319740d65eb3b2a9e80b4a9f989083099"  # 成功commit
+    end_commit = "81dc24b20d9a016dd4e92e76d2849cc9e3d0f8c8"  # 失败commit
     commits = get_commits(start=start_commit, end=end_commit)
     save_pickle(data=commits, filename="candidate_commits.pickle")
     print("the candidate commits is {}".format(commits))
@@ -178,8 +179,8 @@ def _commit_locate(self, commits):
     final_commit = BinarySearch(
         commit_list=commits,
         title="PrecisionBS",
-        layerfile="./layercase/sublayer1000/Clas_cases/CSWinTransformer_CSWinTransformer_base_384/SIR_5.py",
-        testing="yaml/dy^dy2stcinn_eval.yml",
+        layerfile="./layercase/sublayer1000/Det_cases/ppyolo_ppyolo_r50vd_dcn_1x_coco/SIR_58.py",
+        testing="yaml/dy^dy2stcinn_train_inputspec.yml",
         perf_decay=None,  # ["dy2st_eval_cinn_perf", 0.042814, -0.3]
         test_obj=LayerTest,
     )._commit_locate(commits)