-
Notifications
You must be signed in to change notification settings - Fork 150
/
Copy pathmeta.yaml
40 lines (33 loc) · 943 Bytes
/
meta.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package:
name: pytorch-sparse
version: 0.6.18
source:
path: ../..
requirements:
build:
- {{ compiler('c') }} # [win]
host:
- pip
- python {{ environ.get('PYTHON_VERSION') }}
- {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
- {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
run:
- scipy
- pytorch-scatter
- python {{ environ.get('PYTHON_VERSION') }}
- {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
- {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}
build:
string: py{{ environ.get('PYTHON_VERSION').replace('.', '') }}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
script_env:
- FORCE_CUDA
- TORCH_CUDA_ARCH_LIST
- WITH_METIS=1
preserve_egg_dir: True
test:
imports:
- torch_sparse
about:
home: https://github.com/rusty1s/pytorch_sparse
license: MIT
summary: PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations