kernel-6.1: Add additional Neuron module

Add a second neuron module so that we can choose the appropriate driver
based upon the detected hardware. This provides both neuron_2_21 and
neuron_latest in the data directory which is not discoverable by depmod.
The mechanism for loading the driver is driverdog link and load modules
for neuron.

There is now two scripts for checking the neuron driver, one for 2.21.*
and one for latest.

Signed-off-by: Matthew Yeazel <[email protected]>

Author: yeazelm

packages/kernel-6.1/Cargo.toml

@@ -18,9 +18,14 @@ sha512 = "395940a92892d92612c51edc566e9bd7bb59432048a796e9e4934246f20aca8f7cef95
 force-upstream = true
 
 [[package.metadata.build-package.external-files]]
-# Use latest-neuron-srpm-url.sh to get this.
+# Use latest-2.21-neuron-srpm-url.sh to get this.
 url = "https://yum.repos.neuron.amazonaws.com/aws-neuronx-dkms-2.21.37.0.noarch.rpm"
 sha512 = "4bdb01d846f12a04bdbe6902c3008ebadbcb50d4d94e8f81b524c02a4e94da7041cdb5e94fd564e1d9898f52a8cbd960a6a65435b90ba7d7839c807e7ca8736d"
 
+[[package.metadata.build-package.external-files]]
+# Use latest-neuron-srpm.url.sh to get this
+url = "https://yum.repos.neuron.amazonaws.com/aws-neuronx-dkms-2.24.7.0.noarch.rpm"
+sha512 = "2892ad1c6c4ff670d8e4aa0151af57b2b437f647a58ffa58901377b2ddeb504a9e4cce6c148643f8340e1ede8bb26ab66039b0a6bbb184bf5d647c6c71720186"
+
 [build-dependencies]
 microcode = { path = "../microcode" }

packages/kernel-6.1/kernel-6.1.spec

@@ -9,9 +9,11 @@ URL: https://www.kernel.org/
 # Use latest-kernel-srpm-url.sh to get this.
 Source0: https://cdn.amazonlinux.com/al2023/blobstore/f3b09a804dce910af99d51994144a9e73322ac17d13feb47547b70b7833fa40a/kernel-6.1.155-176.282.amzn2023.src.rpm
 Source1: gpgkey-B21C50FA44A99720EAA72F7FE951904AD832C631.asc
-# Use latest-neuron-srpm-url.sh to get this.
+# Use latest-2.21-neuron-srpm-url.sh to get this.
 Source2: https://yum.repos.neuron.amazonaws.com/aws-neuronx-dkms-2.21.37.0.noarch.rpm
-Source3: gpgkey-00FA2C1079260870A76D2C285749CAD8646D9185.asc
+# Use latest-neuron-srpm-url.sh to get this.
+Source3: https://yum.repos.neuron.amazonaws.com/aws-neuronx-dkms-2.24.7.0.noarch.rpm
+Source4: gpgkey-00FA2C1079260870A76D2C285749CAD8646D9185.asc
 
 # Custom Bottlerocket kernel configurations.
 Source100: config-bottlerocket
@@ -30,9 +32,12 @@ Source202: fipsmodules-aarch64
 # Adjust kernel-devel mount behavior if not squashfs.
 Source210: var-lib-kernel-devel-lower.mount.drop-in.conf.in
 
-# Neuron-related drop-ins.
-Source220: neuron-sysinit.target.drop-in.conf
-Source221: [email protected]
+# Neuron-related configuration and unit files
+Source220: neuron-tmpfiles.conf.in
+Source221: neuron-inf1.toml.in
+Source222: neuron-latest.toml.in
+Source223: load-neuron-inf1-modules.service
+Source224: load-neuron-latest-modules.service
 
 # Bootconfig snippets to adjust the default kernel command line for the platform.
 Source300: bootconfig-aws.conf
@@ -252,11 +257,20 @@ rm -f ../config-* ../*.patch
 
 %if "%{_cross_arch}" == "x86_64"
 cd %{_builddir}
-rpmkeys --import %{S:3} --dbpath "${PWD}/rpmdb"
+# 2.21 for inf1 support
+rpmkeys --import %{S:4} --dbpath "${PWD}/rpmdb"
 rpmkeys --checksig %{S:2} --dbpath "${PWD}/rpmdb"
 rm -rf "${PWD}/rpmdb"
 rpm2cpio %{S:2} | cpio -idmu './usr/src/aws-neuronx-*'
-find usr/src/ -mindepth 1 -maxdepth 1 -type d -exec mv {} neuron \;
+find usr/src/ -mindepth 1 -maxdepth 1 -type d -exec mv {} neuron_2_21 \;
+rm -r usr
+
+# latest neuron driver
+rpmkeys --import %{S:4} --dbpath "${PWD}/rpmdb"
+rpmkeys --checksig %{S:3} --dbpath "${PWD}/rpmdb"
+rm -rf "${PWD}/rpmdb"
+rpm2cpio %{S:3} | cpio -idmu './usr/src/aws-neuronx-*'
+find usr/src/ -mindepth 1 -maxdepth 1 -type d -exec mv {} neuron_latest \;
 rm -r usr
 %endif
 
@@ -276,15 +290,21 @@ make -s\\\
 %kmake %{?_smp_mflags} modules
 
 %if "%{_cross_arch}" == "x86_64"
-%kmake %{?_smp_mflags} M=%{_builddir}/neuron
+%kmake %{?_smp_mflags} M=%{_builddir}/neuron_2_21
+%kmake %{?_smp_mflags} M=%{_builddir}/neuron_latest
 %endif
 
 %install
 %kmake %{?_smp_mflags} headers_install
 %kmake %{?_smp_mflags} modules_install
 
 %if "%{_cross_arch}" == "x86_64"
-%kmake %{?_smp_mflags} M=%{_builddir}/neuron modules_install
+install -d %{buildroot}%{_cross_libexecdir}/neuron/neuron_2_21/
+install -d %{buildroot}%{_cross_libexecdir}/neuron/neuron_latest/
+%kmake %{?_smp_mflags} INSTALL_MOD_DIR=neuron_2_21 M=%{_builddir}/neuron_2_21 modules_install
+%kmake %{?_smp_mflags} INSTALL_MOD_DIR=neuron_latest M=%{_builddir}/neuron_latest modules_install
+mv %{buildroot}%{_cross_kmoddir}/neuron_2_21/neuron.ko.gz %{buildroot}%{_cross_libexecdir}/neuron/neuron_2_21/
+mv %{buildroot}%{_cross_kmoddir}/neuron_latest/neuron.ko.gz %{buildroot}%{_cross_libexecdir}/neuron/neuron_latest/
 %endif
 
 install -d %{buildroot}/boot
@@ -426,12 +446,22 @@ sed -e 's|PREFIX|%{_cross_prefix}|g' %{S:210} \
   > %{buildroot}%{_cross_unitdir}/"${LOWERPATH}.mount.d"/no-squashfs.conf
 
 %if "%{_cross_arch}" == "x86_64"
-# Add Neuron-related drop-ins to load the module when the hardware is present.
-mkdir -p %{buildroot}%{_cross_unitdir}/sysinit.target.d
-install -p -m 0644 %{S:220} %{buildroot}%{_cross_unitdir}/sysinit.target.d/neuron.conf
-
-mkdir -p %{buildroot}%{_cross_unitdir}/[email protected]
-install -p -m 0644 %{S:221} %{buildroot}%{_cross_unitdir}/[email protected]/neuron.conf
+# Add Neuron-related configuration files to load the module when the hardware is present.
+install -d 0644 %{buildroot}%{_cross_tmpfilesdir}
+sed \
+  -e "s|__KERNEL_VERSION__|%{version}|" \
+  -e "s|__PREFIX__|%{_cross_prefix}|" %{S:220} > neuron.conf
+install -p -m 0644 neuron.conf %{buildroot}%{_cross_tmpfilesdir}/
+install -d 0644 %{buildroot}%{_cross_factorydir}%{_cross_sysconfdir}/drivers
+# inf1
+sed -e 's|__NEURON_MODULES__|%{_cross_libexecdir}/neuron|' %{S:221} > \
+  neuron-inf1.toml
+install -m 0644 neuron-inf1.toml %{buildroot}%{_cross_factorydir}%{_cross_sysconfdir}/drivers
+# latest
+sed -e 's|__NEURON_MODULES__|%{_cross_libexecdir}/neuron|' %{S:222} > \
+  neuron-latest.toml
+install -m 0644 neuron-latest.toml %{buildroot}%{_cross_factorydir}%{_cross_sysconfdir}/drivers
+install -p -m 0644 %{S:223} %{S:224} %{buildroot}%{_cross_unitdir}
 %endif
 
 # Install platform-specific bootconfig snippets.
@@ -1501,9 +1531,13 @@ install -p -m 0644 %{S:302} %{buildroot}%{_cross_bootconfigdir}/05-metal.conf
 
 %if "%{_cross_arch}" == "x86_64"
 %files modules-neuron
-%{_cross_kmoddir}/extra/neuron.ko.gz
-%{_cross_unitdir}/sysinit.target.d/neuron.conf
-%{_cross_unitdir}/[email protected]/neuron.conf
+%{_cross_libexecdir}/neuron/neuron_2_21/neuron.ko.gz
+%{_cross_libexecdir}/neuron/neuron_latest/neuron.ko.gz
+%{_cross_tmpfilesdir}/neuron.conf
+%{_cross_unitdir}/load-neuron-inf1-modules.service
+%{_cross_unitdir}/load-neuron-latest-modules.service
+%{_cross_factorydir}%{_cross_sysconfdir}/drivers/neuron-inf1.toml
+%{_cross_factorydir}%{_cross_sysconfdir}/drivers/neuron-latest.toml
 %endif
 
 %changelog

packages/kernel-6.1/latest-2.21-neuron-srpms-url.sh

@@ -0,0 +1,9 @@
+#!/bin/sh
+cmd="
+dnf install -q -y --releasever=latest yum-utils &&
+dnf download -q --repofrompath neuron,https://yum.repos.neuron.amazonaws.com --repo=neuron --urls aws-neuronx-dkms-2.21*
+"
+docker run --rm public.ecr.aws/amazonlinux/amazonlinux:2023 sh -c "${cmd}" \
+    | grep '^http' \
+    | xargs --max-args=1 --no-run-if-empty realpath --canonicalize-missing --relative-to=. \
+    | sed 's_:/_://_'

packages/kernel-6.1/load-neuron-inf1-modules.service

@@ -0,0 +1,19 @@
+[Unit]
+Description=Load Neuron Inf1 modules
+After=local-fs.target
+Requires=local-fs.target
+# Disable manual restarts to prevent loading kernel modules
+# that weren't linked by the running system
+RefuseManualStart=true
+RefuseManualStop=true
+
+[Service]
+Type=oneshot
+ExecCondition=/usr/bin/ghostdog match-driver neuron inf1
+ExecStart=/usr/bin/driverdog --modules-set neuron-inf1 link-modules
+ExecStart=/usr/bin/driverdog --modules-set neuron-inf1 load-modules
+RemainAfterExit=true
+StandardError=journal+console
+
+[Install] 
+RequiredBy=drivers.target

packages/kernel-6.1/load-neuron-latest-modules.service

@@ -0,0 +1,19 @@
+[Unit]
+Description=Load Neuron Latest modules
+After=local-fs.target
+Requires=local-fs.target
+# Disable manual restarts to prevent loading kernel modules
+# that weren't linked by the running system
+RefuseManualStart=true
+RefuseManualStop=true
+
+[Service]
+Type=oneshot
+ExecCondition=/usr/bin/ghostdog match-driver neuron latest
+ExecStart=/usr/bin/driverdog --modules-set neuron-latest link-modules
+ExecStart=/usr/bin/driverdog --modules-set neuron-latest load-modules
+RemainAfterExit=true
+StandardError=journal+console
+
+[Install] 
+RequiredBy=drivers.target

packages/kernel-6.1/[email protected]

@@ -0,0 +1,5 @@
+[neuron-inf1]
+lib-modules-path = "kernel/drivers/neuron"
+
+[neuron-inf1.kernel-modules."neuron.ko.gz"]
+copy-source = "__NEURON_MODULES__/neuron_2_21"

packages/kernel-6.1/neuron-inf1.toml.in

@@ -0,0 +1,5 @@
+[neuron-latest]
+lib-modules-path = "kernel/drivers/neuron"
+
+[neuron-latest.kernel-modules."neuron.ko.gz"]
+copy-source = "__NEURON_MODULES__/neuron_latest"

packages/kernel-6.1/neuron-latest.toml.in

@@ -0,0 +1,4 @@
+C /etc/drivers/neuron-inf1.toml
+C /etc/drivers/neuron-latest.toml
+R __PREFIX__/lib/modules/__KERNEL_VERSION__/kernel/drivers/neuron - - - - -
+d __PREFIX__/lib/modules/__KERNEL_VERSION__/kernel/drivers/neuron 0755 root root - -