Commit 1088e0fd authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 6dc2a996
...@@ -76,24 +76,31 @@ benchmark_sdxl: ...@@ -76,24 +76,31 @@ benchmark_sdxl:
stage: test stage: test
tags: tags:
- demos - demos
before_script: # 定义产物,保存测试生成的图片和性能报告
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 ---"
- export LD_LIBRARY_PATH=$CI_PROJECT_DIR/rocblas-install/lib/:$LD_LIBRARY_PATH - 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 LD_LIBRARY_PATH=$CI_PROJECT_DIR/package-miopen-dev-0801-ubuntu20/lib/:$LD_LIBRARY_PATH
- export PYTORCH_MIOPEN_SUGGEST_NHWC=1 - export PYTORCH_MIOPEN_SUGGEST_NHWC=1
- export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH" - export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH"
- export MODEL_PATH="$CI_PROJECT_DIR/stable-diffusion-xl-base-1.0" - 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 "LD_LIBRARY_PATH: $LD_LIBRARY_PATH"
- echo "PYTHONPATH: $PYTHONPATH" - echo "PYTHONPATH: $PYTHONPATH"
- echo "MODEL_PATH: $MODEL_PATH"
# --- 第2部分:运行基准测试 ---
- echo "--- 3. Running SDXL benchmark ---"
- python test.py - python test.py
- echo "--- Benchmark finished ---" - echo "--- Benchmark finished ---"
artifacts: # 只从上一个作业继承缓存,不更新
name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME"
paths:
- "*.png"
- results.json
expire_in: 1 week
cache: cache:
key: "${CI_COMMIT_REF_SLUG}" key: ${CI_COMMIT_REF_SLUG}
policy: pull policy: pull
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment