meta.yaml

package:
  name: pytorch-spline-conv
  version: 1.2.1

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:
    - 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: pip install .
  script_env:
    - FORCE_CUDA
    - TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0 7.5 8.0 8.6+PTX"

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