Skip win (no pytorch), cuda extensions on linux
{% set name = "theseus-ai" %}
{% set version = "0.1.3" %}
+{% set processor = "cpu" if cuda_compiler_version == "None" else "gpu" %} # [not osx]
+{% set processor = "cpu" %} # [osx]
+
+# Prioritize gpu build if cudatoolkit can be installed (through __cuda virtual package)
+{% set build = build + 100 if processor == "gpu" else build %}
+
package:
name: {{ name|lower }}
version: {{ version }}
@@ -12,10 +18,12 @@ source:
build:
script: {{ PYTHON }} -m pip install . -vv
number: 0
+ skip: true # [(osx and cuda_compiler_version != "None") or win]
requirements:
build:
- {{ compiler('cxx') }}
+ - {{ compiler('cuda') }} # [not osx and cuda_compiler_version != "None"]
host:
- python
- pytorch
@@ -28,6 +36,7 @@ requirements:
- pytest >=6.2.1
- pybind11 >=2.7.1
- functorch
+ - __cuda >={{ cuda_compiler_version }} # [not osx and cuda_compiler_version != "None"]
test:
imports:
