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

Update .gitlab-ci.yml

parent 5c22c9b0
...@@ -76,13 +76,11 @@ benchmark_sdxl: ...@@ -76,13 +76,11 @@ benchmark_sdxl:
stage: test stage: test
tags: tags:
- demos - demos
# 在脚本执行前设置所有必要的环境变量
before_script: before_script:
- 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"
# 设置模型路径环境变量,供 test.py 使用
- 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: script:
- echo "--- 3. Running SDXL benchmark ---" - echo "--- 3. Running SDXL benchmark ---"
...@@ -90,15 +88,12 @@ benchmark_sdxl: ...@@ -90,15 +88,12 @@ benchmark_sdxl:
- echo "PYTHONPATH: $PYTHONPATH" - echo "PYTHONPATH: $PYTHONPATH"
- python test.py - python test.py
- echo "--- Benchmark finished ---" - echo "--- Benchmark finished ---"
# 定义产物,保存测试生成的图片和性能报告
artifacts: artifacts:
name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME" name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME"
paths: paths:
- "*.png" - "*.png"
- results.json - results.json
# 产物保存 1 周
expire_in: 1 week 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