Skip to content

Commit ed23ae1

Browse files
committed
Modify experimental workflow
1 parent 612fac1 commit ed23ae1

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

.github/workflows/build-wheel-rocm-windows.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@ jobs:
2525
curl -LO https://download.amd.com/developer/eula/rocm-hub/AMD-Software-PRO-Edition-23.Q3-Win10-Win11-For-HIP.exe
2626
Start-Process 'AMD-Software-PRO-Edition-23.Q3-Win10-Win11-For-HIP.exe' -ArgumentList '-install' -NoNewWindow -Wait
2727
echo "C:\Program Files\AMD\ROCm\5.5\bin" >> $env:GITHUB_PATH
28-
echo "ROCM_PATH=C:\Program Files\AMD\ROCm\5.5" >> $env:GITHUB_ENV
28+
echo 'ROCM_PATH="C:\Program Files\AMD\ROCm\5.5"' >> $env:GITHUB_ENV
29+
echo 'HIP_PATH="C:\Program Files\AMD\ROCm\5.5"' >> $env:GITHUB_ENV
2930
echo "ROCM_VERSION=5.5.1" >> $env:GITHUB_ENV
3031
3132
- uses: actions/setup-python@v3
@@ -38,9 +39,11 @@ jobs:
3839
3940
- name: Build Wheel
4041
run: |
42+
$env:CC = 'C:\Program Files\AMD\ROCm\5.5\bin\clang.exe'
43+
$env:CXX = 'C:\Program Files\AMD\ROCm\5.5\bin\clang++.exe'
4144
$env:VERBOSE = '1'
4245
$env:FORCE_CMAKE = '1'
43-
$env:CMAKE_ARGS = '-DLLAMA_HIPBLAS=on -DCMAKE_HIP_ARCHITECTURES=gfx1030;gfx1100 -DCMAKE_PREFIX_PATH="C:\Program Files\AMD\ROCm\5.5"'
46+
$env:CMAKE_ARGS = "-DLLAMA_HIPBLAS=ON -DCMAKE_HIP_ARCHITECTURES=gfx1030;gfx1100 -DCMAKE_PREFIX_PATH=`"${env:ROCM_PATH}`""
4447
python3 -m build -n --wheel -C--build-option=egg_info "-C--build-option=--tag-build=+rocm5.5.1"
4548
4649
- uses: actions/upload-artifact@v3

0 commit comments

Comments
 (0)