Skip to content

Commit

Permalink
Merge pull request conda-forge#11761 from rluria14/pytorch_sparse
Browse files Browse the repository at this point in the history
add pytorch_sparse
  • Loading branch information
scopatz committed Jun 1, 2020
2 parents 7bf3d92 + ffb5758 commit ddc2d57
Showing 1 changed file with 76 additions and 0 deletions.
76 changes: 76 additions & 0 deletions recipes/torch-sparse/meta.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
{% set name = "torch_sparse" %}
{% set version = "0.6.1" %}

{% set pytorch_sparse_proc_type = "cpu" if cuda_compiler_version == "None" else "gpu" %} # [linux or win]
{% set pytorch_sparse_proc_type = "cpu" %} # [osx]

package:
name: py{{ name|lower }}
version: {{ version }}

source:
url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/{{ name }}-{{ version }}.tar.gz
sha256: 41de6bb39450fc59917c582caa353505d443809af3601c67e4fbaa2284fd1899

build:
number: 0
skip: true # [win]

outputs:
# A meta-package to select CPU or GPU pytorch_sparse build.
- name: pytorch_sparse_proc
version: 1.0.0
build:
number: 0
string: "{{ pytorch_sparse_proc_type }}"
test:
commands:
- exit 0

- name: pytorch_sparse
version: {{ version }}
build:
number: 0
script: python -m pip install . -vv
requirements:
build:
- {{ compiler('c') }}
- {{ compiler('cxx') }}
- {{ compiler('cuda') }} # [linux and cuda_compiler_version != "None"]
host:
- python
- pytorch
- setuptools
- scipy
- pytest-runner
- pip
run:
- python
- pytorch
run_constrained:
- pytorch_sparse_proc * {{ pytorch_sparse_proc_type }}
test:
requires:
- pytest
- pytest-cov
imports:
- torch_sparse
about:
home: https://github.com/rusty1s/pytorch_sparse
summary: PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
license: MIT
license_file: LICENSE
dev_url: https://github.com/rusty1s/pytorch_sparse

about:
home: https://github.com/rusty1s/pytorch_sparse
summary: PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
license: MIT
license_file: LICENSE
dev_url: https://github.com/rusty1s/pytorch_sparse

extra:
recipe-maintainers:
- rluria14
- oblute
- ndmaxar

0 comments on commit ddc2d57

Please sign in to comment.