From f38719b6a3c1af05fdb577bc8cf2b1f6bff15e55 Mon Sep 17 00:00:00 2001 From: Alex Domingo Date: Thu, 12 Feb 2026 23:23:10 +0100 Subject: [PATCH] adding easyconfigs: PyTorch-2.9.1-foss-2025a-whl.eb, PyTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb, Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb --- ...yTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb | 48 +++++++++++++++++++ .../p/PyTorch/PyTorch-2.9.1-foss-2025a-whl.eb | 45 +++++++++++++++++ ...Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb | 40 ++++++++++++++++ 3 files changed, 133 insertions(+) create mode 100644 easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb create mode 100644 easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-whl.eb create mode 100644 easybuild/easyconfigs/t/Triton/Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb diff --git a/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb new file mode 100644 index 000000000000..01452464197d --- /dev/null +++ b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-CUDA-12.8.0-whl.eb @@ -0,0 +1,48 @@ +easyblock = 'PythonBundle' + +name = 'PyTorch' +version = '2.9.1' +versionsuffix = '-CUDA-%(cudaver)s-whl' + +homepage = 'https://pytorch.org/' +description = """PyTorch is an optimized tensor library for deep learning using GPUs and CPUs. +Used for applications such as computer vision, deep learning research and +natural language processing. +This module provides the official wheels of PyTorch from pytorch.org""" + +toolchain = {'name': 'foss', 'version': '2025a'} + +_torch_wheel_pyver = 'cp%(pymajver)s%(pyminver)s-cp%(pymajver)s%(pyminver)s' +_torch_wheel_source = { + 'aarch64': f'torch-{version}%2Bcu128-{_torch_wheel_pyver}-manylinux_2_28_aarch64.whl', + 'x86_64': f'torch-{version}%2Bcu128-{_torch_wheel_pyver}-manylinux_2_28_x86_64.whl' +} + +dependencies = [ + ('CUDA', '12.8.0', '', SYSTEM), + ('CUDA-wheel-bundle', '12.8.0', '-CUDA-%(cudaver)s'), + ('Python', '3.13.1'), + ('Triton', '3.5.1', versionsuffix), + ('SciPy-bundle', '2025.06'), + ('networkx', '3.5'), +] + +exts_list = [ + ('sympy', '1.14.0', { + 'source_urls': ['https://download.pytorch.org/whl'], + 'sources': [SOURCE_PY3_WHL], + 'checsums': ['e091cc3e99d2141a0ba2847328f5479b05d94a6635cb96148ccb3f34671bd8f5'], + }), + ('torch', version, { + 'source_urls': ['https://download.pytorch.org/whl/cu128'], + 'sources': [_torch_wheel_source[ARCH]], + 'checksums': [{ + 'torch-2.9.1%2Bcu128-cp313-cp313-manylinux_2_28_aarch64.whl': + '0b80b7555dcd0a75b7b06016991f01281a0bb078cf28fa2d1dfb949fad2fbd07', + 'torch-2.9.1%2Bcu128-cp313-cp313-manylinux_2_28_x86_64.whl': + '63381a109a569b280ed3319da89d3afe5cf9ab5c879936382a212affb5c90552', + }], + }), +] + +moduleclass = 'ai' diff --git a/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-whl.eb b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-whl.eb new file mode 100644 index 000000000000..13aa9b4ed973 --- /dev/null +++ b/easybuild/easyconfigs/p/PyTorch/PyTorch-2.9.1-foss-2025a-whl.eb @@ -0,0 +1,45 @@ +easyblock = 'PythonBundle' + +name = 'PyTorch' +version = '2.9.1' +versionsuffix = '-whl' + +homepage = 'https://pytorch.org/' +description = """PyTorch is an optimized tensor library for deep learning using GPUs and CPUs. +Used for applications such as computer vision, deep learning research and +natural language processing. +This module provides the official wheels of PyTorch from pytorch.org""" + +toolchain = {'name': 'foss', 'version': '2025a'} + +_torch_wheel_pyver = 'cp%(pymajver)s%(pyminver)s-cp%(pymajver)s%(pyminver)s' +_torch_wheel_source = { + 'aarch64': f'torch-{version}%2Bcpu-{_torch_wheel_pyver}-manylinux_2_28_aarch64.whl', + 'x86_64': f'torch-{version}%2Bcpu-{_torch_wheel_pyver}-manylinux_2_28_x86_64.whl' +} + +dependencies = [ + ('Python', '3.13.1'), + ('SciPy-bundle', '2025.06'), + ('networkx', '3.5'), +] + +exts_list = [ + ('sympy', '1.14.0', { + 'source_urls': ['https://download.pytorch.org/whl'], + 'sources': [SOURCE_PY3_WHL], + 'checsums': ['e091cc3e99d2141a0ba2847328f5479b05d94a6635cb96148ccb3f34671bd8f5'], + }), + ('torch', version, { + 'source_urls': ['https://download.pytorch.org/whl/cpu'], + 'sources': [_torch_wheel_source[ARCH]], + 'checksums': [{ + 'torch-2.9.1%2Bcpu-cp313-cp313-manylinux_2_28_aarch64.whl': + '3e532e553b37ee859205a9b2d1c7977fd6922f53bbb1b9bfdd5bdc00d1a60ed4', + 'torch-2.9.1%2Bcpu-cp313-cp313-manylinux_2_28_x86_64.whl': + '39b3dff6d8fba240ae0d1bede4ca11c2531ae3b47329206512d99e17907ff74b', + }], + }), +] + +moduleclass = 'ai' diff --git a/easybuild/easyconfigs/t/Triton/Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb b/easybuild/easyconfigs/t/Triton/Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb new file mode 100644 index 000000000000..4848a1f4125b --- /dev/null +++ b/easybuild/easyconfigs/t/Triton/Triton-3.5.1-gfbf-2025a-CUDA-12.8.0-whl.eb @@ -0,0 +1,40 @@ +easyblock = 'PythonBundle' + +name = 'Triton' +version = '3.5.1' +versionsuffix = '-CUDA-%(cudaver)s-whl' + +homepage = 'https://triton-lang.org/' +description = """Triton is a language and compiler for parallel programming. It aims to provide a +Python-based programming environment for productively writing custom DNN compute +kernels capable of running at maximal throughput on modern GPU hardware. +This module provides the official wheels of Triton from pytorch.org""" + +toolchain = {'name': 'gfbf', 'version': '2025a'} + +_torch_wheel_pyver = 'cp%(pymajver)s%(pyminver)s-cp%(pymajver)s%(pyminver)s' +_torch_wheel_source = { + 'aarch64': f'triton-{version}-{_torch_wheel_pyver}-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl', + 'x86_64': f'triton-{version}-{_torch_wheel_pyver}-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl' +} + +dependencies = [ + ('CUDA', '12.8.0', '', SYSTEM), + ('Python', '3.13.1'), + ('SciPy-bundle', '2025.06'), +] + +exts_list = [ + ('triton', version, { + 'source_urls': ['https://download.pytorch.org/whl'], + 'sources': [_torch_wheel_source[ARCH]], + 'checksums': [{ + 'triton-3.5.1-cp313-cp313-manylinux_2_27_aarch64.manylinux_2_28_aarch64.whl': + '34e02e5406194a7071a703c9fddde133a590369a670ad81a9b4a0e98320617a9', + 'triton-3.5.1-cp313-cp313-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl': + 'e5cfa13cb273cee1043fbf9cb61c4438deb16d57b37b2c277b945ef185890c66', + }], + }), +] + +moduleclass = 'ai'