{ai}[foss/2025a] PyTorch v2.9.1, Triton v3.5.1 w/ CUDA 12.8.0 from wheels

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'

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'

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'