diff options
Diffstat (limited to 'var/spack/repos/builtin/packages/cutensor/package.py')
-rw-r--r-- | var/spack/repos/builtin/packages/cutensor/package.py | 30 |
1 files changed, 15 insertions, 15 deletions
diff --git a/var/spack/repos/builtin/packages/cutensor/package.py b/var/spack/repos/builtin/packages/cutensor/package.py index 14734bd922..1bb77a82c1 100644 --- a/var/spack/repos/builtin/packages/cutensor/package.py +++ b/var/spack/repos/builtin/packages/cutensor/package.py @@ -9,11 +9,11 @@ from spack.package import * _versions = { # cuTensor 1.5.0 - '1.5.0.3': { - 'Linux-x86_64': '4fdebe94f0ba3933a422cff3dd05a0ef7a18552ca274dd12564056993f55471d', - 'Linux-ppc64le': 'ad736acc94e88673b04a3156d7d3a408937cac32d083acdfbd8435582cbe15db', - 'Linux-aarch64': '5b9ac479b1dadaf40464ff3076e45f2ec92581c07df1258a155b5bcd142f6090'}, - + "1.5.0.3": { + "Linux-x86_64": "4fdebe94f0ba3933a422cff3dd05a0ef7a18552ca274dd12564056993f55471d", + "Linux-ppc64le": "ad736acc94e88673b04a3156d7d3a408937cac32d083acdfbd8435582cbe15db", + "Linux-aarch64": "5b9ac479b1dadaf40464ff3076e45f2ec92581c07df1258a155b5bcd142f6090", + }, } @@ -24,23 +24,23 @@ class Cutensor(Package): homepage = "https://developer.nvidia.com/cutensor" - maintainers = ['bvanessen'] - url = "cutensor" + maintainers = ["bvanessen"] + url = "cutensor" for ver, packages in _versions.items(): key = "{0}-{1}".format(platform.system(), platform.machine()) pkg = packages.get(key) cutensor_ver = ver - cuda_ver = '10.0' - if platform.machine() == 'aarch64': - cuda_ver = '11.0' + cuda_ver = "10.0" + if platform.machine() == "aarch64": + cuda_ver = "11.0" if pkg: version(cutensor_ver, sha256=pkg) # Add constraints matching CUDA version to cuTensor version - cuda_req = 'cuda@{0}:'.format(cuda_ver) - cutensor_ver_req = '@{0}'.format(cutensor_ver) + cuda_req = "cuda@{0}:".format(cuda_ver) + cutensor_ver_req = "@{0}".format(cutensor_ver) depends_on(cuda_req, when=cutensor_ver_req) def url_for_version(self, version): @@ -48,10 +48,10 @@ class Cutensor(Package): sys = "{0}-{1}".format(platform.system(), platform.machine()) # Munge it to match Nvidia's naming scheme sys_key = sys.lower() - sys_key = sys_key.replace('aarch64', 'sbsa') + sys_key = sys_key.replace("aarch64", "sbsa") - url = 'https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/{0}/libcutensor-{0}-{1}-archive.tar.xz' + url = "https://developer.download.nvidia.com/compute/cutensor/redist/libcutensor/{0}/libcutensor-{0}-{1}-archive.tar.xz" return url.format(sys_key, version) def install(self, spec, prefix): - install_tree('.', prefix) + install_tree(".", prefix) |