plt update data api, test=model

Zeref996 · Nov 11, 2024 · de2ae0e · de2ae0e
1 parent 5d00ada
commit de2ae0e
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 1 deletion.
diff --git a/framework/e2e/PaddleLT_new/run.py b/framework/e2e/PaddleLT_new/run.py
@@ -391,7 +391,8 @@ def _test_run(self, py_list):
         error_list = []
         error_count = 0
         for py_file in py_list:
-            self.logger.get_log().info(get_nv_memory(int(os.environ.get("PLT_DEVICE_ID"))))
+            if os.environ.get("PLT_GET_NV_MEMORY") == "True":
+                self.logger.get_log().info(get_nv_memory(int(os.environ.get("PLT_DEVICE_ID"))))
             _py_file, _exit_code = self._single_pytest_run(py_file=py_file, testing=self.testing)
             if _exit_code is not None:
                 error_list.append(_py_file)

diff --git a/framework/e2e/PaddleLT_new/scene/set_benchmark_pts_env.sh b/framework/e2e/PaddleLT_new/scene/set_benchmark_pts_env.sh
@@ -9,6 +9,7 @@ export PLT_MD5="${PLT_MD5:-0a0a0a0a0a0a0a0a0a0a0a0a0a0a0a0a}"
 export PLT_PERF_MODE="${PLT_PERF_MODE:-none}" # 设定性能测试方式, 可选参数: unit-python(独立python执行方式)
 export PLT_PERF_CONTENT="${PLT_PERF_CONTENT:-layer}" # 设定性能测试内容, 可选参数: layer(端到端子图), kernel(nv工具kernel分析)
 export PLT_SET_DEVICE="${PLT_SET_DEVICE:-gpu}" # 硬件
+export PLT_GET_NV_MEMORY="${PLT_GET_NV_MEMORY:-True}"
 export PLT_DEVICE_ID="${PLT_DEVICE_ID:-6}"  # 设备编号
 export CUDA_VISIBLE_DEVICES="${PLT_DEVICE_ID:-6}"
 export FRAMEWORK="${FRAMEWORK:-paddle}"  # 框架种类

diff --git a/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train^dy2stcinn_eval_inputspec_env.sh b/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train^dy2stcinn_eval_inputspec_env.sh
@@ -7,6 +7,7 @@ export CASE_DIR=sublayer1000
 export TESTING=yaml/CI_dy^dy2stcinn_train^dy2stcinn_eval_inputspec.yml # 设定测试项目配置yaml
 export TESTING_MODE=precision_multi_gpu # 设定测试模型为精度
 export PLT_SET_DEVICE=gpu
+export PLT_GET_NV_MEMORY=False
 export PLT_DEVICE_ID="${PLT_DEVICE_ID:-0,1}"
 export CUDA_VISIBLE_DEVICES="${PLT_DEVICE_ID:-0,1}"
 export FRAMEWORK=paddle

diff --git a/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train_env.sh b/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train_env.sh
@@ -7,6 +7,7 @@ export CASE_DIR=sublayer1000
 export TESTING=yaml/CI_dy^dy2stcinn_train.yml # 设定测试项目配置yaml
 export TESTING_MODE=precision_multi_gpu # 设定测试模型为精度
 export PLT_SET_DEVICE=gpu
+export PLT_GET_NV_MEMORY=False
 export PLT_DEVICE_ID="${PLT_DEVICE_ID:-0,1}"
 export CUDA_VISIBLE_DEVICES="${PLT_DEVICE_ID:-0,1}"
 export FRAMEWORK=paddle

diff --git a/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train_inputspec_env.sh b/framework/e2e/PaddleLT_new/scene/set_ci_dy^dy2stcinn_train_inputspec_env.sh
@@ -7,6 +7,7 @@ export CASE_DIR=sublayer1000
 export TESTING=yaml/CI_dy^dy2stcinn_train_inputspec.yml # 设定测试项目配置yaml
 export TESTING_MODE=precision_multi_gpu # 设定测试模型为精度
 export PLT_SET_DEVICE=gpu
+export PLT_GET_NV_MEMORY=False
 export PLT_DEVICE_ID="${PLT_DEVICE_ID:-0,1}"
 export CUDA_VISIBLE_DEVICES="${PLT_DEVICE_ID:-0,1}"
 export FRAMEWORK=paddle

diff --git a/framework/e2e/PaddleLT_new/scene/set_pts_env.sh b/framework/e2e/PaddleLT_new/scene/set_pts_env.sh
@@ -8,6 +8,7 @@ export TESTING_MODE="${TESTING_MODE:-precision}" # 设定测试模型为精度
 export PLT_MD5="${PLT_MD5:-0a0a0a0a0a0a0a0a0a0a0a0a0a0a0a0a}"
 export PLT_PERF_CONTENT="${PLT_PERF_CONTENT:-layer}"
 export PLT_SET_DEVICE="${PLT_SET_DEVICE:-gpu}"
+export PLT_GET_NV_MEMORY="${PLT_GET_NV_MEMORY:-True}"
 export PLT_DEVICE_ID="${PLT_DEVICE_ID:-6}"
 export CUDA_VISIBLE_DEVICES="${PLT_DEVICE_ID:-6}"
 export FRAMEWORK="${FRAMEWORK:-paddle}"