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

Update .gitlab-ci.yml

parent 6dc2a996
......@@ -76,24 +76,31 @@ benchmark_sdxl:
stage: test
tags:
- 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/package-miopen-dev-0801-ubuntu20/lib/:$LD_LIBRARY_PATH
- export PYTORCH_MIOPEN_SUGGEST_NHWC=1
- export PYTHONPATH="$CI_PROJECT_DIR/transformers:$PYTHONPATH"
- 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
expire_in: 1 week
# 只从上一个作业继承缓存,不更新
cache:
key: "${CI_COMMIT_REF_SLUG}"
key: ${CI_COMMIT_REF_SLUG}
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