From 17ae8444ca7f40ff9b257893a9bb81043fce6791 Mon Sep 17 00:00:00 2001
From: Aaron Nelson <aanelson@google.com>
Date: Wed, 25 Feb 2026 10:06:01 -0800
Subject: [PATCH] set_multiqueue: make x4 "multinic accelerator"; extend IDPF
 ring length beyond a4x-max

Makes the previously a4x-max-specific IDPF ring length change apply to
any machine type in the "multinic accelerator" list. Given the curent
list, the only machines this applies for are a4x-max, c4x, and x4.

Of those, c4x is only for its IRDMA interfaces, for which we believe
this change will have no impact.

PiperOrigin-RevId: 875217306
---
 src/usr/bin/google_set_multiqueue | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/src/usr/bin/google_set_multiqueue b/src/usr/bin/google_set_multiqueue
index 7f776e3..8c95c7e 100755
--- a/src/usr/bin/google_set_multiqueue
+++ b/src/usr/bin/google_set_multiqueue
@@ -52,8 +52,8 @@ function get_metadata() {
 }
 
 A4X_ALL_CPUS_MASK="0000ffff,ffffffff,ffffffff,ffffffff,ffffffff"
-A4X_RX_RING_LENGTH="2048"
-A4X_TX_RING_LENGTH="1024"
+IDPF_RX_RING_LENGTH="2048"
+IDPF_TX_RING_LENGTH="1024"
 
 function is_decimal_int() {
   [ "${1}" -eq "${1}" ] > /dev/null 2>&1
@@ -179,7 +179,9 @@ function is_multinic_accelerator_platform() {
   || "$machine_type" == *"a4-highgpu-"* \
   || "$machine_type" == *"a4x-highgpu-"* \
   || "$machine_type" == *"a4x-maxgpu-"* \
-  || "$machine_type" == *"c4x-"* ]] || return 1
+  || "$machine_type" == *"c4x-"* \
+  || "$machine_type" == *"x4-"* \
+  ]] || return 1
   return 0
 }
 
@@ -508,16 +510,13 @@ for ((node=0; node<num_numa_nodes; node++)); do
       continue
     fi
 
-    # For IDPF at least, we need to reconfigure ring size before setting IRQ affinity,
-    # since adjusting the ring size destroys the IRQ affinity settings.
-    if [[ $machine_type == *"a4x-maxgpu-"* ]]; then
-      ethtool -G "$nic_name" rx "$A4X_RX_RING_LENGTH" tx "$A4X_TX_RING_LENGTH"
-    fi
-
     # For non-gvnic/idpf devices (e.g. mlx5), the IRQ bindings will be handled by the device's driver.
     if is_gvnic "$nic_name"; then
       bind_cores_index=$(set_irq_range_gve "$nic_name" "$bind_cores_index" "${node_irq_ranges[@]}")
     elif is_idpf "$nic_name"; then
+      # We need to reconfigure ring size before setting IRQ affinity, because
+      # adjusting the ring size destroys the IRQ affinity settings.
+      ethtool -G "$nic_name" rx "$IDPF_RX_RING_LENGTH" tx "$IDPF_TX_RING_LENGTH"
       bind_cores_index=$(set_irq_range_idpf "$nic_name" "$bind_cores_index" "${node_irq_ranges[@]}")
     else
       echo "$nic_name is not a gvnic/idpf device, not setting irq affinity on this device"