From 90d00f8a599ade2d43a5977788639a7952ad1840 Mon Sep 17 00:00:00 2001 From: Thomas Bouvier Date: Wed, 25 Jan 2023 01:26:04 +0100 Subject: Add package `py-nvidia-dali` (#35058) * Add package `py-nvidia-dali` * Add linux check * Add missing deps --- .../builtin/packages/py-nvidia-dali/package.py | 56 ++++++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 var/spack/repos/builtin/packages/py-nvidia-dali/package.py (limited to 'var') diff --git a/var/spack/repos/builtin/packages/py-nvidia-dali/package.py b/var/spack/repos/builtin/packages/py-nvidia-dali/package.py new file mode 100644 index 0000000000..5e16d0b489 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-nvidia-dali/package.py @@ -0,0 +1,56 @@ +# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other +# Spack Project Developers. See the top-level COPYRIGHT file for details. +# +# SPDX-License-Identifier: (Apache-2.0 OR MIT) + +import platform + +from spack.package import * + + +class PyNvidiaDali(PythonPackage): + """A GPU-accelerated library containing highly optimized building blocks and + an execution engine for data processing to accelerate deep learning training + and inference applications.""" + + homepage = "https://developer.nvidia.com/dali" + url = "https://developer.download.nvidia.com/compute/redist/" + + maintainers = ["thomas-bouvier"] + + system = platform.system().lower() + arch = platform.machine() + if "linux" in system and arch == "x86_64": + version( + "1.22.0-cuda120", + sha256="6cbd9e3139d4c203f61f960f5ad1fc4b461621a60b7fa7ef0ba6d77c780b35f4", + expand=False, + url="https://developer.download.nvidia.com/compute/redist/nvidia-dali-cuda120/nvidia_dali_cuda120-1.22.0-6971317-py3-none-manylinux2014_x86_64.whl", + ) + version( + "1.22.0-cuda110", + sha256="8c3ccc7eddc1f63d3f858448c5c384ab129273e0c140e091aca2a98d48c5a80c", + expand=False, + preferred=True, + url="https://developer.download.nvidia.com/compute/redist/nvidia-dali-cuda110/nvidia_dali_cuda110-1.22.0-6988993-py3-none-manylinux2014_x86_64.whl", + ) + elif "linux" in system and arch == "aarch64": + version( + "1.22.0-cuda120", + sha256="5e496eebeba3bc1cddd18e081c8c45121283478931cbe9b64912d2394d0942ca", + expand=False, + url="https://developer.download.nvidia.com/compute/redist/nvidia-dali-cuda120/nvidia_dali_cuda120-1.22.0-6971317-py3-none-manylinux2014_aarch64.whl", + ) + version( + "1.22.0-cuda110", + sha256="0da47629fec01abf418fda0eeb393998820e40f6fae6b4c7d3e625aa4cdba6bd", + expand=False, + preferred=True, + url="https://developer.download.nvidia.com/compute/redist/nvidia-dali-cuda110/nvidia_dali_cuda110-1.22.0-6988993-py3-none-manylinux2014_aarch64.whl", + ) + + depends_on("python@3.6:3.10", type=("build", "run")) + depends_on("cuda@12", when="@1.22.0-cuda120", type=("build", "run")) + depends_on("cuda@11", when="@1.22.0-cuda110", type=("build", "run")) + depends_on("py-astunparse@1.6.0:", type=("build", "run")) + depends_on("py-gast@0.2.1:0.4.0", type=("build", "run")) -- cgit v1.2.3-70-g09d2