diff --git a/framework/e2e/PaddleLT_new/run.py b/framework/e2e/PaddleLT_new/run.py index 6b904e2d2b..af80ad7036 100644 --- 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 index f511896e14..c4a022fd95 100644 --- 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 index 1f647b6985..00bd3d0f5f 100644 --- 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 index 3f358d6374..419d9d1c22 100644 --- 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 index ffe6b02ab3..c7f3eda6dd 100644 --- 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 index e092e1ebc6..6ec0127062 100644 --- 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}"