Commit 6dc2a996 authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 5c22c9b0
......@@ -76,13 +76,11 @@ benchmark_sdxl:
stage: test
tags:
- demos
# 在脚本执行前设置所有必要的环境变量
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 ---"
......@@ -90,15 +88,12 @@ benchmark_sdxl:
- echo "PYTHONPATH: $PYTHONPATH"
- 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}
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