Skip to content

Commit

Permalink
update sdk version for windows cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
vinovo committed Oct 25, 2024
1 parent e884290 commit 74c44d4
Show file tree
Hide file tree
Showing 2 changed files with 51 additions and 48 deletions.
96 changes: 48 additions & 48 deletions .github/workflows/build-wheels-cpu.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -92,59 +92,59 @@ jobs:
name: wheels-${{ matrix.os }}
path: ./wheelhouse/*.whl

# build_wheels_arm64:
# name: Build arm64 wheels
# runs-on: ubuntu-20.04
# steps:
# - uses: actions/checkout@v4
# with:
# submodules: "recursive"
build_wheels_arm64:
name: Build arm64 wheels
runs-on: ubuntu-20.04
steps:
- uses: actions/checkout@v4
with:
submodules: "recursive"

# - name: Set up QEMU
# uses: docker/setup-qemu-action@v3
# with:
# platforms: linux/arm64
- name: Set up QEMU
uses: docker/setup-qemu-action@v3
with:
platforms: linux/arm64

# - name: Build wheels
# uses: pypa/[email protected]
# env:
# CIBW_REPAIR_WHEEL_COMMAND: ""
# CIBW_BUILD_FRONTEND: "build"
# CIBW_SKIP: "*musllinux* pp*"
# CIBW_ARCHS: "aarch64"
# CIBW_BUILD: "cp37-* cp38-* cp39-* cp310-* cp311-* cp312-* cp313-*"
# CMAKE_BUILD_PARALLEL_LEVEL: $(nproc)
# with:
# output-dir: wheelhouse
- name: Build wheels
uses: pypa/[email protected]
env:
CIBW_REPAIR_WHEEL_COMMAND: ""
CIBW_BUILD_FRONTEND: "build"
CIBW_SKIP: "*musllinux* pp*"
CIBW_ARCHS: "aarch64"
CIBW_BUILD: "cp37-* cp38-* cp39-* cp310-* cp311-* cp312-* cp313-*"
CMAKE_BUILD_PARALLEL_LEVEL: $(nproc)
with:
output-dir: wheelhouse

# - name: Upload wheels as artifacts
# uses: actions/upload-artifact@v4
# with:
# name: wheels_arm64
# path: ./wheelhouse/*.whl
- name: Upload wheels as artifacts
uses: actions/upload-artifact@v4
with:
name: wheels_arm64
path: ./wheelhouse/*.whl

# build_sdist:
# name: Build source distribution
# runs-on: ubuntu-latest
build_sdist:
name: Build source distribution
runs-on: ubuntu-latest

# steps:
# - uses: actions/checkout@v4
# with:
# submodules: "recursive"
# - uses: actions/setup-python@v5
# with:
# python-version: "3.8"
# - name: Install dependencies
# run: |
# python -m pip install --upgrade pip build
# python -m pip install -e .
# - name: Build source distribution
# run: |
# python -m build --sdist
# - uses: actions/upload-artifact@v4
# with:
# name: sdist
# path: ./dist/*.tar.gz
steps:
- uses: actions/checkout@v4
with:
submodules: "recursive"
- uses: actions/setup-python@v5
with:
python-version: "3.8"
- name: Install dependencies
run: |
python -m pip install --upgrade pip build
python -m pip install -e .
- name: Build source distribution
run: |
python -m build --sdist
- uses: actions/upload-artifact@v4
with:
name: sdist
path: ./dist/*.tar.gz

# release:
# name: Release
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/build-wheels-cuda-win.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,8 @@ jobs:
with:
submodules: "recursive"

- uses: GuillaumeFalourd/setup-windows10-sdk-action@v2

- uses: actions/setup-python@v5
with:
python-version: ${{ matrix.pyver }}
Expand Down Expand Up @@ -115,6 +117,7 @@ jobs:
$env:CMAKE_ARGS = '-DSD_CUBLAS=ON -DGGML_CUDA=ON -DGGML_CUBLAS=ON -DCMAKE_CUDA_ARCHITECTURES=all'
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DGGML_CUDA_FORCE_MMQ=ON -DGGML_AVX2=OFF -DGGML_FMA=OFF -DGGML_F16C=OFF'
$env:CMAKE_ARGS = $env:CMAKE_ARGS + ' -DCMAKE_SYSTEM_VERSION=10.0.18362.0 -DCMAKE_VS_WINDOWS_TARGET_PLATFORM_VERSION=10.0.18362.0'
$env:CMAKE_BUILD_PARALLEL_LEVEL = $(nproc)
python -m build --wheel
Expand Down

0 comments on commit 74c44d4

Please sign in to comment.