Commit 8cb8d2d2 authored by Paul McCarthy's avatar Paul McCarthy 🚵
Browse files

MNT: Update CUDA VERSIONS, adjust skip platform to use "linux-64-cuda" rather

than separate IDS for each CUDA ver
parent f10d47d7
......@@ -15,7 +15,7 @@ from typing import List
# Note: These lists must be updated whenever new
# platforms / CUDA versions become supported.
COMPILATION_TARGETS = ['linux-64', 'macos-64']
CUDA_VERSIONS = ['9.2', '10.2', '11.0']
CUDA_VERSIONS = ['9.2', '10.2', '11.0', '11.1', '11.3']
def get_platform(meta : dict) -> str:
......@@ -109,10 +109,10 @@ def get_skip_platform(meta : dict) -> str:
"""Generates a value for the FSLCONDA_SKIP_PLATFORM variable. """
pkgtype = get_platform(meta)
binaries = ' '.join(COMPILATION_TARGETS)
cudas = ' '.join([f'linux-64-cuda-{ver}' for ver in CUDA_VERSIONS])
cuda = 'linux-64-cuda'
if pkgtype == 'noarch': return ' '.join((binaries, cudas))
elif pkgtype == 'binary': return ' '.join(('noarch', cudas))
if pkgtype == 'noarch': return ' '.join((binaries, cuda))
elif pkgtype == 'binary': return ' '.join(('noarch', cuda))
elif pkgtype == 'cuda': return ' '.join(('noarch', binaries))
raise ValueError(f'Unknown platform {pkgtype}')
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment