Commit 2da00a7e authored by rusty1s's avatar rusty1s
Browse files

[skip ci] build conda

parent 4d976c5c
...@@ -16,7 +16,7 @@ jobs: ...@@ -16,7 +16,7 @@ jobs:
# torch-version: [1.8.0, 1.9.0] # torch-version: [1.8.0, 1.9.0]
torch-version: [1.9.0] torch-version: [1.9.0]
# cuda-version: ['cpu', 'cu101', 'cu102', 'cu111'] # cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
cuda-version: ['cpu'] cuda-version: ['cpu', 'cu102']
exclude: exclude:
- torch-version: 1.9.0 - torch-version: 1.9.0
cuda-version: 'cu101' cuda-version: 'cu101'
...@@ -52,7 +52,7 @@ jobs: ...@@ -52,7 +52,7 @@ jobs:
- name: Build Conda package for CPU - name: Build Conda package for CPU
if: ${{ matrix.cuda-version == 'cpu' }} if: ${{ matrix.cuda-version == 'cpu' }}
run: | run: |
FORCE_ONLY_CPU=1 ./conda/pytorch-spline-conv/build_conda.sh ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }} FORCE_CUDA=0 ./conda/pytorch-spline-conv/build_conda.sh ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }}
shell: shell:
bash bash
......
...@@ -22,7 +22,6 @@ build: ...@@ -22,7 +22,6 @@ build:
script_env: script_env:
- CUDA_HOME - CUDA_HOME
- FORCE_CUDA - FORCE_CUDA
- FORCE_ONLY_CPU
test: test:
imports: imports:
......
...@@ -19,6 +19,8 @@ if os.getenv('FORCE_ONLY_CPU', '0') == '1': ...@@ -19,6 +19,8 @@ if os.getenv('FORCE_ONLY_CPU', '0') == '1':
BUILD_DOCS = os.getenv('BUILD_DOCS', '0') == '1' BUILD_DOCS = os.getenv('BUILD_DOCS', '0') == '1'
print('FORCE CUDA', suffices)
def get_extensions(): def get_extensions():
extensions = [] extensions = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment