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

[skip ci] build conda

parent 4d976c5c
......@@ -16,7 +16,7 @@ jobs:
# torch-version: [1.8.0, 1.9.0]
torch-version: [1.9.0]
# cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
cuda-version: ['cpu']
cuda-version: ['cpu', 'cu102']
exclude:
- torch-version: 1.9.0
cuda-version: 'cu101'
......@@ -52,7 +52,7 @@ jobs:
- name: Build Conda package for CPU
if: ${{ matrix.cuda-version == 'cpu' }}
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:
bash
......
......@@ -22,7 +22,6 @@ build:
script_env:
- CUDA_HOME
- FORCE_CUDA
- FORCE_ONLY_CPU
test:
imports:
......
......@@ -19,6 +19,8 @@ if os.getenv('FORCE_ONLY_CPU', '0') == '1':
BUILD_DOCS = os.getenv('BUILD_DOCS', '0') == '1'
print('FORCE CUDA', suffices)
def get_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