meta.yaml 883 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
package:
rusty1s's avatar
rusty1s committed
2
3
  name: pytorch-spline-conv
  version: 1.2.1
rusty1s's avatar
rusty1s committed
4
5

source:
rusty1s's avatar
rusty1s committed
6
  path: ../..
rusty1s's avatar
rusty1s committed
7
8

requirements:
rusty1s's avatar
rusty1s committed
9
10
11
  build:
    - {{ compiler('c') }}  # [win]

rusty1s's avatar
rusty1s committed
12
  host:
rusty1s's avatar
rusty1s committed
13
    - python {{ environ.get('PYTHON_VERSION') }}
rusty1s's avatar
rusty1s committed
14
15
16
17
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

  run:
rusty1s's avatar
rusty1s committed
18
    - python {{ environ.get('PYTHON_VERSION') }}
rusty1s's avatar
rusty1s committed
19
20
21
22
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

build:
rusty1s's avatar
rusty1s committed
23
  string: py{{ environ.get('PYTHON_VERSION').replace('.', '') }}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
rusty1s's avatar
rusty1s committed
24
  script: python setup.py install
rusty1s's avatar
rusty1s committed
25
26
  script_env:
    - FORCE_CUDA
rusty1s's avatar
rusty1s committed
27
28
29
30
31
32
33
34
35

test:
  imports:
    - torch_spline_conv

about:
  home: https://github.com/rusty1s/pytorch_spline_conv
  license: MIT
  summary: Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch