Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -76,30 +76,28 @@ benchmark_sdxl:
   stage: test
   tags:
     - demos
-  # 定义产物，保存测试生成的图片和性能报告
-  artifacts:
-    name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME"
-    paths:
-      - "*.png"
-      - results.json
-    expire_in: 1 week
-  # 将所有命令都放在 script 中，结构更清晰
-  script:
-    # --- 第1部分：设置环境变量 ---
-    - echo "--- 3. Setting up environment variables ---"
+  # 在脚本执行前设置所有必要的环境变量
+  before_script:
     - export LD_LIBRARY_PATH=$CI_PROJECT_DIR/rocblas-install/lib/:$LD_LIBRARY_PATH
     - export LD_LIBRARY_PATH=$CI_PROJECT_DIR/package-miopen-dev-0801-ubuntu20/lib/:$LD_LIBRARY_PATH
     - export PYTORCH_MIOPEN_SUGGEST_NHWC=1
     - export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH"
+    # 设置模型路径环境变量，供 test.py 使用
     - export MODEL_PATH="$CI_PROJECT_DIR/stable-diffusion-xl-base-1.0"
+  script:
+    - echo "--- 3. Running SDXL benchmark ---"
     - echo "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"
     - echo "PYTHONPATH: $PYTHONPATH"
-    - echo "MODEL_PATH: $MODEL_PATH"
-    
-    # --- 第2部分：运行基准测试 ---
-    - echo "--- 3. Running SDXL benchmark ---"
     - python test.py
     - echo "--- Benchmark finished ---"
+  # 定义产物，保存测试生成的图片和性能报告
+  artifacts:
+    name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME"
+    paths:
+      - "*.png"
+      - results.json
+    # 产物保存 1 周
+    expire_in: 1 week
   # 只从上一个作业继承缓存，不更新
   cache:
     key: ${CI_COMMIT_REF_SLUG}