kmod-6.12-nvidia-r570: Add IMEX

This adds IMEX as a subpackage that provides support for GB200 devices
which require the IMEX process to coordinate efficient communication
between devices in ComputeDomains.

Signed-off-by: Matthew Yeazel <yeazelm@amazon.com>

Author: Matthew Yeazel

packages/kmod-6.12-nvidia-r570/Cargo.toml

@@ -36,6 +36,16 @@ url = "https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/nvidi
 sha512 = "c6425cb53bb7915f5912e49c2b4182fe9875eb60759c15e130ae6d323dc168ad860451e2d69932aba8febb350bc7c0ff6d82df625aa0b63d8601e9fa7cc31b2d"
 force-upstream = true
 
+[[package.metadata.build-package.external-files]]
+url = "https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/nvidia-imex-570-570.148.08-1.x86_64.rpm"
+sha512 = "846e3f2c6dab2bf097771ec01720cf98ccd30f0dcbad09826d708e29f896982642bad866c38fc116c02f6f0a3feea3eaef69034380dc303dbb50aeee079ed01a"
+force-upstream = true
+
+[[package.metadata.build-package.external-files]]
+url = "https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/nvidia-imex-570-570.148.08-1.aarch64.rpm"
+sha512 = "13c1ac2f8650a909ea826fbd41c4ba3378a6c83aa6d38996c2a5509aa49f70a90c023ccaa1e1cad6ecd6808e721cc4effcf02f4cc21f0f2a0fc4043a105ba1bd"
+force-upstream = true
+
 [[package.metadata.build-package.external-files]]
 url = "https://raw.githubusercontent.com/NVIDIA/open-gpu-kernel-modules/570/COPYING"
 sha512 = "f9cee68cbb12095af4b4e92d01c210461789ef41c70b64efefd6719d0b88468b7a67a3629c432d4d9304c730b5d1a942228a5bcc74a03ab1c411c77c758cd938"

packages/kmod-6.12-nvidia-r570/kmod-6.12-nvidia-r570.spec

@@ -39,6 +39,10 @@ Source3: COPYING
 Source10: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/nvidia-fabric-manager-%{tesla_ver}-1.x86_64.rpm
 Source11: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/nvidia-fabric-manager-%{tesla_ver}-1.aarch64.rpm
 
+# IMEX for GB200
+Source20: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/nvidia-imex-%{tesla_ver}-1.x86_64.rpm
+Source21: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/nvidia-imex-%{tesla_ver}-1.aarch64.rpm
+
 # Common NVIDIA conf files from 200 to 299
 Source200: nvidia-tmpfiles.conf.in
 Source202: nvidia-dependencies-modules-load.conf
@@ -85,10 +89,18 @@ Requires: %{name}-grid
 Summary: NVIDIA fabricmanager config and service files
 Requires: %{name}-tesla(fabricmanager)
 Requires: %{_cross_os}nvlsm
+Requires: %{name}-imex
 
 %description fabricmanager
 %{summary}.
 
+%package imex
+Summary: NVIDIA IMEX config and service files
+Requires: %{name}
+
+%description imex
+%{summary}.
+
 %package open-gpu
 Summary: NVIDIA %{tesla_major} Open GPU driver
 Version: %{tesla_ver}
@@ -141,6 +153,11 @@ rpm2cpio %{_sourcedir}/nvidia-fabric-manager-%{tesla_ver}-1.%{_cross_arch}.rpm |
 # Add the license.
 install -p -m 0644 %{S:2} %{S:3} .
 
+# Extract fabricmanager from the rpm via cpio rather than `%%setup` since the
+# correct source is architecture-dependent.
+mkdir imex-%{fm_arch}-%{tesla_ver}-archive
+rpm2cpio %{_sourcedir}/nvidia-imex-%{tesla_major}-%{tesla_ver}-1.%{_cross_arch}.rpm | cpio -idmV -D imex-%{fm_arch}-%{tesla_ver}-archive
+
 # This recipe was based in the NVIDIA yum/dnf specs:
 # https://github.com/NVIDIA/yum-packaging-precompiled-kmod
 
@@ -442,6 +459,13 @@ done
 
 popd
 
+# Begin IMEX binaries and configuration files
+pushd imex-%{fm_arch}-%{tesla_ver}-archive
+install -p -m 0755 usr/bin/nvidia-imex %{buildroot}%{_cross_bindir}
+install -p -m 0755 usr/bin/nvidia-imex-ctl %{buildroot}%{_cross_bindir}
+
+popd
+
 %files
 %{_cross_attribution_file}
 %dir %{_cross_libexecdir}/nvidia
@@ -708,3 +732,8 @@ popd
 %{_cross_factorydir}%{_cross_sysconfdir}/nvidia/fabricmanager.cfg
 %{_cross_factorydir}%{_cross_sysconfdir}/nvidia/fabricmanager.env
 %{_cross_unitdir}/nvidia-fabricmanager.service
+
+%files imex
+%license imex-%{fm_arch}-%{tesla_ver}-archive/usr/share/licenses/nvidia-imex-%{tesla_major}/LICENSE
+%{_cross_bindir}/nvidia-imex
+%{_cross_bindir}/nvidia-imex-ctl