set_multiqueue: set a4x-max ring len before irq affinity

PiperOrigin-RevId: 859268025

Author: a-r-n

src/usr/bin/google_set_multiqueue

@@ -508,14 +508,17 @@ 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
       bind_cores_index=$(set_irq_range_idpf "$nic_name" "$bind_cores_index" "${node_irq_ranges[@]}")
-      if [[ $machine_type == *"a4x-maxgpu-"* ]]; then
-        ethtool -G "$nic_name" rx "$A4X_RX_RING_LENGTH" tx "$A4X_TX_RING_LENGTH"
-      fi
     else
       echo "$nic_name is not a gvnic/idpf device, not setting irq affinity on this device"
     fi