{tools}[GCCcore/14.2.0] PyTorch v2.9.1, parameterized v0.9.0, pytest-flakefinder v1.1.0, ... w/ CUDA 12.8.0

easybuild/easyconfigs/e/expecttest/expecttest-0.3.0-GCCcore-14.2.0.eb

@@ -0,0 +1,25 @@
+easyblock = 'PythonPackage'
+
+name = 'expecttest'
+version = '0.3.0'
+
+homepage = 'https://github.com/ezyang/expecttest'
+description = """This library implements expect tests (also known as "golden" tests). Expect tests are a method of
+ writing tests where instead of hard-coding the expected output of a test, you run the test to get the output, and
+ the test framework automatically populates the expected output. If the output of the test changes, you can rerun
+ the test with the environment variable EXPECTTEST_ACCEPT=1 to automatically update the expected output."""
+
+toolchain = {'name': 'GCCcore', 'version': '14.2.0'}
+
+sources = [SOURCE_TAR_GZ]
+checksums = ['6e8512fb86523ada1f94fd1b14e280f924e379064bb8a29ee399950e513eeccd']
+
+builddependencies = [
+    ('binutils', '2.42'),
+    ('poetry', '2.1.2'),
+]
+dependencies = [
+    ('Python', '3.13.1'),
+]
+
+moduleclass = 'tools'

easybuild/easyconfigs/n/NVSHMEM/NVSHMEM-3.3.20-gompi-2025a-CUDA-12.8.0.eb

@@ -0,0 +1,63 @@
+easyblock = 'CMakeMake'
+
+name = 'NVSHMEM'
+version = '3.3.20'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://developer.nvidia.com/nvshmem'
+description = """NVSHMEM is a parallel programming interface based on OpenSHMEM that provides
+efficient and scalable communication for NVIDIA GPU clusters. NVSHMEM creates a
+global address space for data that spans the memory of multiple GPUs and can be
+accessed with fine-grained GPU-initiated operations, CPU-initiated operations,
+and operations on CUDA streams.
+"""
+
+toolchain = {'name': 'gompi', 'version': '2025a'}
+
+source_urls = [
+    'https://github.com/NVIDIA/nvshmem/releases/download/v%(version)s-0/',
+    'https://developer.download.nvidia.com/compute/redist/nvshmem/%(version)s/source/',
+]
+sources = ['%(namelower)s_src_cuda12-all-all-%(version)s.tar.gz']
+patches = ['NVSHMEM-3.3.20_update_cxx_standard.patch']
+
+checksums = [
+    # nvshmem_src_cuda12-all-all-3.3.20.tar.gz
+    '96ec9620e82ec90de92c7d61a7ba03c0eba05075bf10e1fc4a066d45e7f7d21f',
+    # NVSHMEM-3.3.20_update_cxx_standard.patch
+    '560eda0fb6e44c8f7666fb18a87d5b6505f0fb77316908718df6e835db52b49f'
+]
+
+builddependencies = [
+    ('Autotools', '20240712'),
+    ('pkgconf', '2.3.0'),
+    ('CMake', '3.31.3'),
+]
+
+dependencies = [
+    ('CUDA', '12.8.0', '', SYSTEM),
+    ('NCCL', '2.27.7', versionsuffix),
+    ('UCX-CUDA', '1.18.0', versionsuffix),
+]
+
+configopts = '-DNVSHMEM_USE_GDRCOPY=1 '
+configopts += '-DGDRCOPY_HOME=${EBROOTGDRCOPY} '
+configopts += '-DMPI_HOME=${EBROOTOPENMPI} '
+configopts += '-DNVSHMEM_MPI_SUPPORT=1 '
+configopts += '-DNVSHMEMTEST_USE_MPI_LAUNCHER=1 '
+configopts += '-DNCCL_HOME=${EBROOTNCCL} '
+configopts += '-DNVSHMEM_USE_NCCL=1 '
+# configopts += '-DNVSHMEM_IBGDA_SUPPORT=1 '
+configopts += '-DNVSHMEM_PREFIX=%(installdir)s '
+# NVSHMEM builds a wheel package if this option is enabled.
+# Properly installing the Python bindings is better handled in a separate EC.
+configopts += '-DNVSHMEM_BUILD_PYTHON_LIB=OFF '
+
+sanity_check_paths = {
+    'files': ['lib/libnvshmem.a', 'lib/nvshmem_bootstrap_mpi.%s' % SHLIB_EXT],
+    'dirs': ['include']
+}
+
+modextravars = {'NVSHMEM_HOME': '%(installdir)s'}
+
+moduleclass = 'devel'

easybuild/easyconfigs/o/optree/optree-0.18.0-GCCcore-14.2.0.eb

@@ -0,0 +1,24 @@
+easyblock = 'PythonPackage'
+
+name = 'optree'
+version = '0.18.0'
+
+homepage = 'https://optree.readthedocs.io/en/latest/'
+description = "Optimized PyTree Utilities"
+
+toolchain = {'name': 'GCCcore', 'version': '14.2.0'}
+
+sources = [SOURCE_TAR_GZ]
+checksums = ['3804fb6ddc923855db2dc4805b4524c66e00f1ef30b166be4aadd52822b13e06']
+
+builddependencies = [
+    ('binutils', '2.42'),
+    ('CMake', '3.31.3'),
+]
+dependencies = [
+    ('Python', '3.13.1'),
+]
+
+preinstallopts = "sed -i '/^license/d' pyproject.toml && "
+
+moduleclass = 'lib'