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

Update .gitlab-ci.yml

parent f55162d6
......@@ -68,18 +68,24 @@ benchmark_sdxl:
- demos
script:
- 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/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"
- 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 "Last 50 lines of the benchmark log:"
- tail -n 50 benchmark.log # 在主日志中显示最后50行,方便快速查看
artifacts:
name: "$CI_JOB_NAME-$CI_COMMIT_REF_NAME"
paths:
- "*.png"
- "results.json"
- "benchmark.log" # <-- 关键:将日志文件也作为产物保存
expire_in: 1 week
cache:
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