zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
2c51b5853f
spack/var/spack/repos/builtin/packages/nvhpc/package.py

614 lines
27 KiB
Python
Raw Blame History

# Copyright Spack Project Developers. See COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
#
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
import platform
from spack.package import *
# FIXME Remove hack for polymorphic versions
# This package uses a ugly hack to be able to dispatch, given the same
# version, to different binary packages based on the platform that is
# running spack. See #13827 for context.
# If you need to add a new version, please be aware that:
# - versions in the following dict are automatically added to the package
# - version tuple must be in the form (checksum, url)
# - checksum must be sha256
# - package key must be in the form '{os}-{arch}' where 'os' is in the
# format returned by platform.system() and 'arch' by platform.machine()
_versions = {
"25.1": {
"Linux-aarch64": (
"0e1d694d54d44559155024d5bab4ca6764eba52d3f27b89f5c252416976e0360",
"https://developer.download.nvidia.com/hpc-sdk/25.1/nvhpc_2025_251_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"0813791f8363f4c493db7891b00396ce522cb73910279b8f18a440aedda6727c",
"https://developer.download.nvidia.com/hpc-sdk/25.1/nvhpc_2025_251_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.11": {
"Linux-aarch64": (
"f2f64e5dec5e90dad5e12a31a992172b0aa19abf872ef1c54a1a437c7008eefb",
"https://developer.download.nvidia.com/hpc-sdk/24.11/nvhpc_2024_2411_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"0c27d66ed0e2d3007d30ac904922a9abf96475197dc0f4dcc6316d235a1dc0c3",
"https://developer.download.nvidia.com/hpc-sdk/24.11/nvhpc_2024_2411_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.9": {
"Linux-aarch64": (
"8d900f798ef806c64993fd4fedf2c2c812dd1ccdbac2a0d33fabcd0cd36f19cf",
"https://developer.download.nvidia.com/hpc-sdk/24.9/nvhpc_2024_249_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"30c493350cf67481e84cea60a3a869e01fa0bcb71df8e898266273fbdf0a7f26",
"https://developer.download.nvidia.com/hpc-sdk/24.9/nvhpc_2024_249_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.7": {
"Linux-aarch64": (
"256ae392ed961162f3f6dc633498db2b68441103a6192f5d4a1c18fa96e992e7",
"https://developer.download.nvidia.com/hpc-sdk/24.7/nvhpc_2024_247_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"bf2094aa2fc5bdbcbf9bfa0fddc1cbed1bfa6e9342980649db2350d9f675f853",
"https://developer.download.nvidia.com/hpc-sdk/24.7/nvhpc_2024_247_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.5": {
"Linux-aarch64": (
"c52b5ba370e053472cbffb825ba1da5b6abaee93d4e15479ec12c32d6ebc47d5",
"https://developer.download.nvidia.com/hpc-sdk/24.5/nvhpc_2024_245_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"e26c5027ffd83fd9e854946670a97253e950cdbacd4894a6715aea91070042ae",
"https://developer.download.nvidia.com/hpc-sdk/24.5/nvhpc_2024_245_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.3": {
"Linux-aarch64": (
"6385847de5f8725e5c56d2abf70c90fed5490f2e71a7bd13d3f4ada8720ef036",
"https://developer.download.nvidia.com/hpc-sdk/24.3/nvhpc_2024_243_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"a9fe5ec878e9c4cc332de732c6739f97ac064ce76ad3d0af6d282658d27124cb",
"https://developer.download.nvidia.com/hpc-sdk/24.3/nvhpc_2024_243_Linux_x86_64_cuda_multi.tar.gz",
),
},
"24.1": {
"Linux-aarch64": (
"8c2ce561d5901a03eadce7f07dce5fbc55e8e88c87b74cf60e01e2eca231c41c",
"https://developer.download.nvidia.com/hpc-sdk/24.1/nvhpc_2024_241_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"e7330eb35e23dcd9b0b3bedc67c0d5443c4fd76b59caa894a76ecb0d17f71f43",
"https://developer.download.nvidia.com/hpc-sdk/24.1/nvhpc_2024_241_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"27992e5fd56af8738501830daddc5e9510ebd553326fea8730236fee4f0f1dd8",
"https://developer.download.nvidia.com/hpc-sdk/24.1/nvhpc_2024_241_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.11": {
"Linux-aarch64": (
"cf744498d1d74ba0af4294388706644ad3669eb0cacea3b69e23739afa2806a0",
"https://developer.download.nvidia.com/hpc-sdk/23.11/nvhpc_2023_2311_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"b08591438bd0802d4c7c78e7c5bc36383a59591b8c2fa8aed5c4b87b24f7bfbb",
"https://developer.download.nvidia.com/hpc-sdk/23.11/nvhpc_2023_2311_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"33483a069a911f9309cd53859ab90d2778fb176df906e9e8d2bd55f45eeec400",
"https://developer.download.nvidia.com/hpc-sdk/23.11/nvhpc_2023_2311_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.9": {
"Linux-aarch64": (
"dd32ae4233438adb71b2b4f8891f04802fdf90f67036ecf18bfde1b6043a03c3",
"https://developer.download.nvidia.com/hpc-sdk/23.9/nvhpc_2023_239_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"984d61695499db098fd32be8345c1f7d7c637ea3bdb29cef17aad656f16b000f",
"https://developer.download.nvidia.com/hpc-sdk/23.9/nvhpc_2023_239_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"ecf343ecad2398e21c8d7f24a580b2932348017dfd8ea38c1ef31b37114b2d4b",
"https://developer.download.nvidia.com/hpc-sdk/23.9/nvhpc_2023_239_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.7": {
"Linux-aarch64": (
"d3b9b674045e6e17156b298941be4e1e1e7dea6a3c1938f14ad653b180860ff2",
"https://developer.download.nvidia.com/hpc-sdk/23.7/nvhpc_2023_237_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"67b137cf67e2c8556ef3952d1ee35f4966c9d1968626825924fb8e4b198a532b",
"https://developer.download.nvidia.com/hpc-sdk/23.7/nvhpc_2023_237_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"fea91d95ff18bca1ce7afde50371caa02001ade8bed6ddfc5ff70862ccbebece",
"https://developer.download.nvidia.com/hpc-sdk/23.7/nvhpc_2023_237_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.5": {
"Linux-aarch64": (
"3af202ad36bbf205b2af56aabe63b971c01b5ec0e82a02effb3c4928f63bc657",
"https://developer.download.nvidia.com/hpc-sdk/23.5/nvhpc_2023_235_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"65b1f1780175096fa58e98e80fc897c0d563cacce43b3e87ba157f40a8e34877",
"https://developer.download.nvidia.com/hpc-sdk/23.5/nvhpc_2023_235_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"071d7119006cb1d7ac22cb91338c20133a02d394efe14931dfa6f5d7dfa54c81",
"https://developer.download.nvidia.com/hpc-sdk/23.5/nvhpc_2023_235_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.3": {
"Linux-aarch64": (
"48c837de0b1d2dc31c313b19da752d27527e706cb4150c7a6185a4218fc24ef3",
"https://developer.download.nvidia.com/hpc-sdk/23.3/nvhpc_2023_233_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"0f2b5b2fff8ffed13428c55809841b4abd3dfe61e2b1a866bbe959e4bcd50223",
"https://developer.download.nvidia.com/hpc-sdk/23.3/nvhpc_2023_233_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"b94d2b4ae5a23c1a0af7d5b07c785f5057850fe3a6ee5ba0aacdde1019af5d12",
"https://developer.download.nvidia.com/hpc-sdk/23.3/nvhpc_2023_233_Linux_x86_64_cuda_multi.tar.gz",
),
},
"23.1": {
"Linux-aarch64": (
"5b430e03752954ea62ac1c745b1735cfdaa43b2e981a9412c1465ecb0412fff6",
"https://developer.download.nvidia.com/hpc-sdk/23.1/nvhpc_2023_231_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"81759e7c747bf4f552b75e7657301f76ecc0828b94fe860f81108c6e83e6ad2b",
"https://developer.download.nvidia.com/hpc-sdk/23.1/nvhpc_2023_231_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"55a064415f6d4ce6a01823ee27ebd266f4fb579679871e7c1a7c054bdc18e9f5",
"https://developer.download.nvidia.com/hpc-sdk/23.1/nvhpc_2023_231_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.11": {
"Linux-aarch64": (
"e60e798657c33b06754d33dfd5ab3bea2882d4a9b9476102303edf2bbe3b7a95",
"https://developer.download.nvidia.com/hpc-sdk/22.11/nvhpc_2022_2211_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"ef800203cf6040b3a5df24f19944b272f62caee8362875bcb394e86dc1de2353",
"https://developer.download.nvidia.com/hpc-sdk/22.11/nvhpc_2022_2211_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"cb91b3a04368457d5cfe3c0e9c0611591fdc8076b01ea977343fe7db7fdcfa3c",
"https://developer.download.nvidia.com/hpc-sdk/22.11/nvhpc_2022_2211_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.9": {
"Linux-aarch64": (
"bc4473f04b49bc9a26f08c17a72360650ddf48a3b6eefacdc525d79c8d730f30",
"https://developer.download.nvidia.com/hpc-sdk/22.9/nvhpc_2022_229_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"9aac31d36bb09f6653544978021f5b78c272112e7748871566f7e930f5e7475b",
"https://developer.download.nvidia.com/hpc-sdk/22.9/nvhpc_2022_229_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"aebfeb826ace3dabf9699f72390ca0340f8789a8ef6fe4032e3c7b794f073ea3",
"https://developer.download.nvidia.com/hpc-sdk/22.9/nvhpc_2022_229_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.7": {
"Linux-aarch64": (
"2aae3fbfd2d0d2d09448a36166c42311368f5600c7c346f159c280b412fe924a",
"https://developer.download.nvidia.com/hpc-sdk/22.7/nvhpc_2022_227_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"6dd4fd382c22769e4fa9508714119abd7d1df3dc58c69414a14b0b0dbc34564f",
"https://developer.download.nvidia.com/hpc-sdk/22.7/nvhpc_2022_227_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"3ce1c346f8bc7e50defb41c545c8907fdc012ff60b27eb8985cf3213f19d863a",
"https://developer.download.nvidia.com/hpc-sdk/22.7/nvhpc_2022_227_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.5": {
"Linux-aarch64": (
"ceeee84e6227e973ad1beded6008d330e3790f7c4598b948fa530fedfa830a16",
"https://developer.download.nvidia.com/hpc-sdk/22.5/nvhpc_2022_225_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"54d1e45664352d0f9f85ab476dd39496dd1b290e0e1221d3bf63afb940dbe16d",
"https://developer.download.nvidia.com/hpc-sdk/22.5/nvhpc_2022_225_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"7674bcf7a77570fafee5b8299959c9e998a9a10bb27904335cf1a58b71766137",
"https://developer.download.nvidia.com/hpc-sdk/22.5/nvhpc_2022_225_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.3": {
"Linux-aarch64": (
"e0ea1cbb726556f6879f4b5dfe17238f8e7680c772368577945a85c0e08328f0",
"https://developer.download.nvidia.com/hpc-sdk/22.3/nvhpc_2022_223_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"5e80db6010adc85fe799dac961ae69e43fdf18d35243666c96a70ecdb80bd280",
"https://developer.download.nvidia.com/hpc-sdk/22.3/nvhpc_2022_223_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"bc60a6faf2237bf20550718f71079a714563fa85df62c341cb833f70eb2fe7bb",
"https://developer.download.nvidia.com/hpc-sdk/22.3/nvhpc_2022_223_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.2": {
"Linux-aarch64": (
"a8241d1139a768d9a0066d1853748160e4098253024e17e997983884d0d33a19",
"https://developer.download.nvidia.com/hpc-sdk/22.2/nvhpc_2022_222_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"f84f72423452968d5bbe02e297f188682c4759864a736a72b32acb3433db3a26",
"https://developer.download.nvidia.com/hpc-sdk/22.2/nvhpc_2022_222_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"8dfb4007d6912b2722946358ac69409592c1f03426d81971ffbcb6fc5fea2cb8",
"https://developer.download.nvidia.com/hpc-sdk/22.2/nvhpc_2022_222_Linux_x86_64_cuda_multi.tar.gz",
),
},
"22.1": {
"Linux-aarch64": (
"05cfa8c520a34eab01272a261b157d421a9ff7129fca7d859b944ce6a16d2255",
"https://developer.download.nvidia.com/hpc-sdk/22.1/nvhpc_2022_221_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"9fa9b64fba2c9b287b5800693417d8065c695d18cab0526bad41d9aecc8be2b3",
"https://developer.download.nvidia.com/hpc-sdk/22.1/nvhpc_2022_221_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"7e4366509ed9031ff271e73327dd3121909902a81ac436307801a5373efaff5e",
"https://developer.download.nvidia.com/hpc-sdk/22.1/nvhpc_2022_221_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.11": {
"Linux-aarch64": (
"3b11bcd9cca862fabfce1e7bcaa2050ea12130c7e897f4e7859ba4c155d20720",
"https://developer.download.nvidia.com/hpc-sdk/21.11/nvhpc_2021_2111_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"ac51ed92de4eb5e1bdb064ada5bbace5b89ac732ad6c6473778edfb8d29a6527",
"https://developer.download.nvidia.com/hpc-sdk/21.11/nvhpc_2021_2111_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"d8d8ccd0e558d22bcddd955f2233219c96f7de56aa8e09e7be833e384d32d6aa",
"https://developer.download.nvidia.com/hpc-sdk/21.11/nvhpc_2021_2111_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.9": {
"Linux-aarch64": (
"52c2c66e30043add4afccedf0ba77daa0000bf42e0db844baa630bb635b91a7d",
"https://developer.download.nvidia.com/hpc-sdk/21.9/nvhpc_2021_219_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"cff0b55fb782be1982bfeec1d9763b674ddbf84ff2c16b364495299266320289",
"https://developer.download.nvidia.com/hpc-sdk/21.9/nvhpc_2021_219_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"7de6a6880fd7e59afe0dee51f1fae4d3bff1ca0fb8ee234b24e1f2fdff23ffc9",
"https://developer.download.nvidia.com/hpc-sdk/21.9/nvhpc_2021_219_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.7": {
"Linux-aarch64": (
"73eb3513845b59645f118b1e313472f54519dc252d5f5c32a05df2a2a8a19878",
"https://developer.download.nvidia.com/hpc-sdk/21.7/nvhpc_2021_217_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"37ea23b5a9c696fb3fdb82855643afc4e02aea618102ec801206441f10fc9fba",
"https://developer.download.nvidia.com/hpc-sdk/21.7/nvhpc_2021_217_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"49d6e23492d131474698cf12971722d42e13a54a4eddec382e66e1053b4ac902",
"https://developer.download.nvidia.com/hpc-sdk/21.7/nvhpc_2021_217_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.5": {
"Linux-aarch64": (
"1a1748cd7cf538199d92ab3b1208935fa4a62708ba21125aeadb328ddc7380d4",
"https://developer.download.nvidia.com/hpc-sdk/21.5/nvhpc_2021_215_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"4674931a5ce28724308cb9cebd546eefa3f0646d3d08adbea28ba5ad27f0c163",
"https://developer.download.nvidia.com/hpc-sdk/21.5/nvhpc_2021_215_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"21989e52c58a6914743631c8200de1fec7e10b3449c6c1833f3032ee74b85f8e",
"https://developer.download.nvidia.com/hpc-sdk/21.5/nvhpc_2021_215_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.3": {
"Linux-aarch64": (
"88e0dbf8fcdd06a2ba06aacf65ae1625b8683688f6593ed3bf8ce129ce1b17b7",
"https://developer.download.nvidia.com/hpc-sdk/21.3/nvhpc_2021_213_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"08cd0cd6c80d633f107b44f88685ada7f014fbf6eac19ef5ae4a7952cabe4037",
"https://developer.download.nvidia.com/hpc-sdk/21.3/nvhpc_2021_213_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"391d5604a70f61bdd4ca6a3e4692f6f2391948990c8a35c395b6867341890031",
"https://developer.download.nvidia.com/hpc-sdk/21.3/nvhpc_2021_213_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.2": {
"Linux-aarch64": (
"fe19c0232f7c9534f8699b7432483c9cc649f1e92e7f0961d1aa7c54d83297ff",
"https://developer.download.nvidia.com/hpc-sdk/21.2/nvhpc_2021_212_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"6b69b6e4ebec6a91b9f1627384c50adad79ebdd25dfb20a5f64cf01c3a07f11a",
"https://developer.download.nvidia.com/hpc-sdk/21.2/nvhpc_2021_212_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"a3e3393040185ae844002fbc6c8eb4ffdfb97ce8b2ce29d796fe7e9a521fdc59",
"https://developer.download.nvidia.com/hpc-sdk/21.2/nvhpc_2021_212_Linux_x86_64_cuda_multi.tar.gz",
),
},
"21.1": {
"Linux-aarch64": (
"b276e7c0ff78cee837a597d9136cd1d8ded27a9d1fdae1e7d674e2a072a9a6aa",
"https://developer.download.nvidia.com/hpc-sdk/21.1/nvhpc_2021_211_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"bc236c212097bac6b7d04d627d9cc6b75bb6cd473a0b6a1bf010559ce328a2b0",
"https://developer.download.nvidia.com/hpc-sdk/21.1/nvhpc_2021_211_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"d529daf46404724ac3f005be4239f2c30e53f5220bb9453f367dccc3a74d6b41",
"https://developer.download.nvidia.com/hpc-sdk/21.1/nvhpc_2021_211_Linux_x86_64_cuda_multi.tar.gz",
),
},
"20.11": {
"Linux-aarch64": (
"2f26ca45b07b694b8669e4f761760d4f7faa8d032b21e430adee1af0a27032c1",
"https://developer.download.nvidia.com/hpc-sdk/20.11/nvhpc_2020_2011_Linux_aarch64_cuda_multi.tar.gz",
),
"Linux-ppc64le": (
"99e5a5437e82f3914e0fe81feb761a5b599a3fe8b31f3c2cac8ae47e8cdc7b0f",
"https://developer.download.nvidia.com/hpc-sdk/20.11/nvhpc_2020_2011_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"c80fc26e5ba586696f7030f03054c1aaca0752a891c7923faf47eb23b66857ec",
"https://developer.download.nvidia.com/hpc-sdk/20.11/nvhpc_2020_2011_Linux_x86_64_cuda_multi.tar.gz",
),
},
"20.9": {
"Linux-aarch64": (
"3bfb3d17f5ee99998bcc30d738e818d3b94b828e2d8da7db48bf152a01e22023",
"https://developer.download.nvidia.com/hpc-sdk/20.9/nvhpc_2020_209_Linux_aarch64_cuda_11.0.tar.gz",
),
"Linux-ppc64le": (
"b2966d4047e1dfd981ce63b333ab9c0acbdc2a6a505fa217456ac9fa3b8e7474",
"https://developer.download.nvidia.com/hpc-sdk/20.9/nvhpc_2020_209_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"fe665ab611b03846a90bd70ca4e08c1e59ab527364b971ed0304e0ae73c778d8",
"https://developer.download.nvidia.com/hpc-sdk/20.9/nvhpc_2020_209_Linux_x86_64_cuda_multi.tar.gz",
),
},
"20.7": {
"Linux-aarch64": (
"5b83ca1919199ac0aa609309b31c345c5a6453dd3131fddeef9e3ee9059a0e9b",
"https://developer.download.nvidia.com/hpc-sdk/20.7/nvhpc_2020_207_Linux_aarch64_cuda_11.0.tar.gz",
),
"Linux-ppc64le": (
"800ead240bdf61611910b2f6df24ee1d7359377ff3767c923738dd81fcea9312",
"https://developer.download.nvidia.com/hpc-sdk/20.7/nvhpc_2020_207_Linux_ppc64le_cuda_multi.tar.gz",
),
"Linux-x86_64": (
"a5c5c8726d2210f2310a852c6d6e03c9ef8c75e3643e9c94e24909f5e9c2ea7a",
"https://developer.download.nvidia.com/hpc-sdk/20.7/nvhpc_2020_207_Linux_x86_64_cuda_multi.tar.gz",
),
},
}
class Nvhpc(Package, CompilerPackage):
"""The NVIDIA HPC SDK is a comprehensive suite of compilers, libraries
and tools essential to maximizing developer productivity and the
performance and portability of HPC applications. The NVIDIA HPC
SDK C, C++, and Fortran compilers support GPU acceleration of HPC
modeling and simulation applications with standard C++ and
Fortran, OpenACC directives, and CUDA. GPU-accelerated math
libraries maximize performance on common HPC algorithms, and
optimized communications libraries enable standards-based
multi-GPU and scalable systems programming. Performance profiling
and debugging tools simplify porting and optimization of HPC
applications."""
homepage = "https://developer.nvidia.com/hpc-sdk"
maintainers("samcmill")
tags = ["e4s", "compiler"]
skip_version_audit = ["platform=darwin", "platform=windows"]
redistribute(source=False, binary=False)
for ver, packages in _versions.items():
key = "{0}-{1}".format(platform.system(), platform.machine())
pkg = packages.get(key)
if pkg:
version(ver, sha256=pkg[0], url=pkg[1])
depends_on("c", type="build") # generated
depends_on("cxx", type="build") # generated
depends_on("fortran", type="build") # generated
variant("blas", default=True, description="Enable BLAS")
variant(
"install_type",
default="single",
values=("single", "network"),
multi=False,
description="Network installs are for installations shared "
"by different operating systems",
)
variant("lapack", default=True, description="Enable LAPACK")
variant("mpi", default=False, description="Enable MPI")
variant(
"default_cuda", default="default", description="Default CUDA version, for example 11.8"
)
provides("blas", when="+blas")
provides("lapack", when="+lapack")
provides("mpi", when="+mpi")
requires("%gcc", msg="nvhpc must be installed with %gcc")
# For now we only detect compiler components
# It will require additional work to detect mpi/lapack/blas components
compiler_languages = ["c", "cxx", "fortran"]
c_names = ["nvc"]
cxx_names = ["nvc++"]
fortran_names = ["nvfortran"]
compiler_version_argument = "--version"
compiler_version_regex = r"nv[^ ]* (?:[^ ]+ Dev-r)?([0-9.]+)(?:-[0-9]+)?"
@classmethod
def determine_variants(cls, exes, version_str):
# TODO: use other exes to determine default_cuda/install_type/blas/lapack/mpi variants
return "~blas~lapack~mpi", {"compilers": cls.determine_compiler_paths(exes=exes)}
def _version_prefix(self):
return join_path(self.prefix, "Linux_%s" % self.spec.target.family, self.version)
def setup_build_environment(self, env):
env.set("NVHPC_SILENT", "true")
env.set("NVHPC_ACCEPT_EULA", "accept")
env.set("NVHPC_INSTALL_DIR", self.prefix)
if self.spec.variants["default_cuda"].value != "default":
env.set("NVHPC_DEFAULT_CUDA", self.spec.variants["default_cuda"].value)
if self.spec.variants["install_type"].value == "network":
local_dir = join_path(self._version_prefix(), "share_objects")
env.set("NVHPC_INSTALL_TYPE", "network")
env.set("NVHPC_INSTALL_LOCAL_DIR", local_dir)
else:
env.set("NVHPC_INSTALL_TYPE", "single")
def install(self, spec, prefix):
compilers_bin = join_path(self._version_prefix(), "compilers", "bin")
install = Executable("./install")
makelocalrc = Executable(join_path(compilers_bin, "makelocalrc"))
makelocalrc_args = [
"-gcc",
self.compiler.cc,
"-gpp",
self.compiler.cxx,
"-g77",
self.compiler.f77,
"-x",
compilers_bin,
]
if self.spec.variants["install_type"].value == "network":
local_dir = join_path(self._version_prefix(), "share_objects")
makelocalrc_args.extend(["-net", local_dir])
# Run install script
install()
# Update localrc to use Spack gcc
makelocalrc(*makelocalrc_args)
def setup_run_environment(self, env):
prefix = Prefix(
join_path(self.prefix, "Linux_%s" % self.spec.target.family, self.version, "compilers")
)
env.set("CC", join_path(prefix.bin, "nvc"))
env.set("CXX", join_path(prefix.bin, "nvc++"))
env.set("F77", join_path(prefix.bin, "nvfortran"))
env.set("FC", join_path(prefix.bin, "nvfortran"))
env.prepend_path("PATH", prefix.bin)
env.prepend_path("LIBRARY_PATH", prefix.lib)
env.prepend_path("LD_LIBRARY_PATH", prefix.lib)
env.prepend_path("MANPATH", prefix.man)
if "+mpi" in self.spec:
mpi_prefix = Prefix(
join_path(
self.prefix,
"Linux_%s" % self.spec.target.family,
self.version,
"comm_libs",
"mpi",
)
)
env.prepend_path("PATH", mpi_prefix.bin)
env.prepend_path("LD_LIBRARY_PATH", mpi_prefix.lib)
def setup_dependent_build_environment(self, env, dependent_spec):
prefix = Prefix(
join_path(self.prefix, "Linux_%s" % self.spec.target.family, self.version, "compilers")
)
env.prepend_path("LIBRARY_PATH", prefix.lib)
env.prepend_path("LD_LIBRARY_PATH", prefix.lib)
if "+mpi" in self.spec:
mpi_prefix = Prefix(
join_path(
self.prefix,
"Linux_%s" % self.spec.target.family,
self.version,
"comm_libs",
"mpi",
)
)
env.prepend_path("LD_LIBRARY_PATH", mpi_prefix.lib)
def setup_dependent_package(self, module, dependent_spec):
if "+mpi" in self.spec or self.provides("mpi"):
mpi_prefix = Prefix(
join_path(
self.prefix,
"Linux_%s" % self.spec.target.family,
self.version,
"comm_libs",
"mpi",
)
)
self.spec.mpicc = join_path(mpi_prefix.bin, "mpicc")
self.spec.mpicxx = join_path(mpi_prefix.bin, "mpicxx")
self.spec.mpif77 = join_path(mpi_prefix.bin, "mpif77")
self.spec.mpifc = join_path(mpi_prefix.bin, "mpif90")
@property
def libs(self):
prefix = Prefix(
join_path(self.prefix, "Linux_%s" % self.spec.target.family, self.version, "compilers")
)
libs = []
if "+blas" in self.spec:
libs.append("libblas")
if "+lapack" in self.spec:
libs.append("liblapack")
libs.append("libnvf")
return find_libraries(libs, root=prefix, recursive=True)
# Avoid binding stub libraries by absolute path
non_bindable_shared_objects = ["stubs"]
Reference in New Issue View Git Blame Copy Permalink