Skip to content

Commit 7578ef0

Browse files
committed
Finalize Windows ROCm workflow
1 parent 9723dfc commit 7578ef0

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

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

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,15 @@ jobs:
4545
- uses: actions/upload-artifact@v3
4646
with:
4747
name: 'lib'
48-
path: ./**/llama.dll
48+
path: ./build/bin/llama.dll
4949

5050
build_wheel:
51-
name: Build ROCm Lib
51+
name: Build ROCm Wheels
5252
runs-on: windows-latest
5353
needs: build_libs
54+
strategy:
55+
matrix:
56+
pyver: ["3.8", "3.9", "3.10", "3.11"]
5457
defaults:
5558
run:
5659
shell: pwsh
@@ -68,17 +71,19 @@ jobs:
6871

6972
- uses: actions/setup-python@v3
7073
with:
71-
python-version: "3.10"
74+
python-version: ${{ matrix.pyver }}
7275

7376
- name: Install Dependencies
7477
run: |
7578
python -m pip install build wheel cmake scikit-build ninja
7679
77-
- name: Build Lib
80+
- name: Build Wheel
7881
run: |
82+
$newsetup = (Get-Content 'setup.py' -raw).Replace("packages=[`"llama_cpp`", `"llama_cpp.server`"],","packages=[`"llama_cpp`", `"llama_cpp.server`"],`n package_data={'llama_cpp': ['llama.dll']},")
83+
New-Item 'setup.py' -itemType File -value $newsetup -force
7984
python setup.py --skip-cmake bdist_wheel egg_info --tag-build=+rocm5.5.1
8085
8186
- uses: actions/upload-artifact@v3
8287
with:
83-
name: 'lib'
84-
path: ./**/llama.dll
88+
name: 'wheels'
89+
path: ./dist/*.whl

0 commit comments

Comments
 (0)