"src/include/migraph/reflect.hpp" did not exist on "76f7ae49f71eb0892c0d531a275c371e3ef61e56"
meta.yaml 801 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
package:
  name: torch-spline-conv
  version: 1.1.1

source:
rusty1s's avatar
rusty1s committed
6
  path: ../..
rusty1s's avatar
rusty1s committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

requirements:
  host:
    - pip
    - python
    - numpy>=1.11
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

  run:
    - python
    - numpy>=1.11
    - {{ environ.get('CONDA_PYTORCH_CONSTRAINT') }}
    - {{ environ.get('CONDA_CUDATOOLKIT_CONSTRAINT') }}

build:
  string: py{{py}}_torch_{{ environ['TORCH_VERSION'] }}_{{ environ['CUDA_VERSION'] }}
rusty1s's avatar
rusty1s committed
24
  script: python setup.py install
rusty1s's avatar
rusty1s committed
25
  script_env:
rusty1s's avatar
rusty1s committed
26
    - CUDA_HOME
rusty1s's avatar
rusty1s committed
27
28
    - FORCE_CUDA
    - FORCE_ONLY_CPU
rusty1s's avatar
rusty1s committed
29
30
31
32
33
34
35
36
37

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