Skip to content

Commit

Permalink
debug
Browse files Browse the repository at this point in the history
  • Loading branch information
nickjbrowning committed Nov 1, 2024
1 parent 2647254 commit 22808b3
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion .github/workflows/build-torch-wheels.yml
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,16 @@ jobs:
CIBW_REPAIR_WHEEL_COMMAND_LINUX: |
auditwheel repair --exclude libmetatensor.so --exclude libtorch.so --exclude libtorch_cpu.so --exclude libc10.so -w {dest_dir} {wheel}
- name: Configure PyTorch
run: |
pip install torch==${{ matrix.torch-version }}.*
env:
PIP_INDEX_URL: https://download.pytorch.org/whl/cu124

- name: Install remaining dependencies from pyproject.toml
run: |
pip install . --no-deps
- name: Build sphericart-torch wheels
run: |
# ensure we build the wheel from the sdist, not the checkout
Expand All @@ -95,7 +105,6 @@ jobs:
TORCH_CUDA_ARCH_LIST=All
CUDAARCHS=all
SPHERICART_TORCH_BUILD_WITH_TORCH_VERSION=${{ matrix.torch-version }}.*
PIP_INDEX_URL=https://download.pytorch.org/whl/cu124
CIBW_REPAIR_WHEEL_COMMAND_LINUX: |
auditwheel repair --exclude libtorch.so --exclude libtorch_cpu.so --exclude libtorch_cuda.so --exclude libc10.so --exclude libc10_cuda.so -w {dest_dir} {wheel}
Expand Down

0 comments on commit 22808b3

Please sign in to comment.