Commit faf4ee06 authored by jerrrrry's avatar jerrrrry
Browse files

Update .gitlab-ci.yml

parent 7a85d4dc
......@@ -19,7 +19,7 @@ benchmark_hunyuan-dit:
script:
- echo "========================================="
- echo "Step 1: Setting up the environment for BATCH_SIZE = $BATCH_SIZE"
- echo "Step 1:Setting up the environment for BATCH_SIZE = $BATCH_SIZE"
- echo "========================================="
# 定义预置文件的根目录
- export PACKAGE_DIR="/workspace/packages/hunyuan-dit"
......@@ -28,13 +28,13 @@ benchmark_hunyuan-dit:
- tar -xzf "$PACKAGE_DIR/package_0915_ubuntu.tar.gz"
- echo "========================================="
- echo "Step 2: Installing Python packages from /workspace/packages/hunyuan-dit"
- echo "Step 2:Installing Python packages from /workspace/packages/hunyuan-dit"
- echo "========================================="
# 从预置目录安装 Python 包
- pip install "$PACKAGE_DIR/apex-1.5.0+das.opt1.dtk25041-cp310-cp310-linux_x86_64.whl" "$PACKAGE_DIR/deepspeed-0.14.2+das.opt1.dtk25041-cp310-cp310-manylinux_2_28_x86_64.whl" "$PACKAGE_DIR/lightop-0.5.0+das.dtk25041.unknown-cp310-cp310-linux_x86_64.whl"
- echo "========================================="
- echo "Step 3: Running the benchmark with BATCH_SIZE = $BATCH_SIZE"
- echo "Step 3:Running the benchmark with BATCH_SIZE = $BATCH_SIZE"
- echo "========================================="
# 设置环境变量,模型路径指向预置目录
- export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/hipblaslt-install/lib/:$LD_LIBRARY_PATH
......
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