Update .gitlab-ci.yml

.gitlab-ci.yml

@@ -27,7 +27,8 @@ benchmark_hunyuan-dit:
     # 从预置目录安装 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" 
     - pip install -r requirements.txt   -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
-
+    - export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/hipblaslt-install/lib:$LD_LIBRARY_PATH
+    - export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/package/miopen/lib:$LD_LIBRARY_PATH
     - echo "========================================="
     - echo "Step 3:Starting serial benchmark with a for-loop"
     - echo "========================================="
@@ -46,8 +47,8 @@ benchmark_hunyuan-dit:
         echo "============================================================"
         
         # 设置本次运行的环境变量 (这些变量在每次循环中都需要)
-        export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/hipblaslt-install/lib/:$LD_LIBRARY_PATH
-        export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/package/miopen/lib/:$LD_LIBRARY_PATH
+        export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/hipblaslt-install/lib:$LD_LIBRARY_PATH
+        export LD_LIBRARY_PATH=/workspace/packages/hunyuan-dit/package/miopen/lib:$LD_LIBRARY_PATH
         
         # 运行测试脚本，将 $bs 作为 batch-size 参数
         python sample_t2i_dcu.py \