Jetson AGX Xavier Wifi disabled after waking up from sleep (latest Jetpack 5.1.4)

We have a fix in pcie driver. Please give it a try.

diff --git a/drivers/pci/controller/dwc/pcie-designware-host.c b/drivers/pci/controller/dwc/pcie-designware-host.c
index 57d0a93..0250069 100644
--- a/drivers/pci/controller/dwc/pcie-designware-host.c
+++ b/drivers/pci/controller/dwc/pcie-designware-host.c
@@ -300,6 +300,7 @@
 	struct pci_host_bridge *bridge;
 	struct resource *cfg_res;
 	int ret;
+	u32 ctrl, num_ctrls;
 
 	raw_spin_lock_init(&pci->pp.lock);
 
@@ -383,6 +384,10 @@
 		}
 
 		if (!pp->ops->msi_host_init) {
+			num_ctrls = pp->num_vectors / MAX_MSI_IRQS_PER_CTRL;
+			for (ctrl = 0; ctrl < num_ctrls; ctrl++)
+				pp->irq_mask[ctrl] = ~0;
+
 			pp->msi_irq_chip = &dw_pci_msi_bottom_irq_chip;
 
 			ret = dw_pcie_allocate_domains(pp);
@@ -580,7 +585,6 @@
 
 		/* Initialize IRQ Status array */
 		for (ctrl = 0; ctrl < num_ctrls; ctrl++) {
-			pp->irq_mask[ctrl] = ~0;
 			dw_pcie_writel_dbi(pci, PCIE_MSI_INTR0_MASK +
 					    (ctrl * MSI_REG_CTRL_BLOCK_SIZE),
 					    pp->irq_mask[ctrl]);