Commit 953fd2c5 authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent f55162d6
...@@ -68,18 +68,24 @@ benchmark_sdxl: ...@@ -68,18 +68,24 @@ benchmark_sdxl:
- demos - demos
script: script:
- echo "--- 3. Running SDXL benchmark ---" - echo "--- 3. Running SDXL benchmark ---"
# 设置环境变量
- 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"
- python test.py - echo "Starting benchmark script, full log will be saved to benchmark.log..."
# 将 Python 脚本的所有输出(包括错误)都重定向到 benchmark.log 文件
- python test.py > benchmark.log 2>&1
- echo "--- Benchmark finished ---" - echo "--- Benchmark finished ---"
- echo "Last 50 lines of the benchmark log:"
- tail -n 50 benchmark.log # 在主日志中显示最后50行,方便快速查看
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"
- "benchmark.log" # <-- 关键:将日志文件也作为产物保存
expire_in: 1 week expire_in: 1 week
cache: cache:
key: "assets-cache-${CI_COMMIT_REF_SLUG}" key: "assets-cache-${CI_COMMIT_REF_SLUG}"
......
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