u32 *base_vector);
        void (*get_msi_address)(struct ath11k_base *ab, u32 *msi_addr_lo,
                                u32 *msi_addr_hi);
+       void (*ce_irq_enable)(struct ath11k_base *ab);
+       void (*ce_irq_disable)(struct ath11k_base *ab);
 };
 
+static inline void ath11k_hif_ce_irq_enable(struct ath11k_base *ab)
+{
+       if (ab->hif.ops->ce_irq_enable)
+               ab->hif.ops->ce_irq_enable(ab);
+}
+
+static inline void ath11k_hif_ce_irq_disable(struct ath11k_base *ab)
+{
+       if (ab->hif.ops->ce_irq_disable)
+               ab->hif.ops->ce_irq_disable(ab);
+}
+
 static inline int ath11k_hif_start(struct ath11k_base *sc)
 {
        return sc->hif.ops->start(sc);
 
        }
 }
 
-static void ath11k_pci_stop(struct ath11k_base *ab)
+static void ath11k_pci_ce_irq_disable_sync(struct ath11k_base *ab)
 {
        ath11k_pci_ce_irqs_disable(ab);
        ath11k_pci_sync_ce_irqs(ab);
        ath11k_pci_kill_tasklets(ab);
+}
+
+static void ath11k_pci_stop(struct ath11k_base *ab)
+{
+       ath11k_pci_ce_irq_disable_sync(ab);
        ath11k_ce_cleanup_pipes(ab);
 }
 
        return 0;
 }
 
+static void ath11k_pci_hif_ce_irq_enable(struct ath11k_base *ab)
+{
+       ath11k_pci_ce_irqs_enable(ab);
+}
+
+static void ath11k_pci_hif_ce_irq_disable(struct ath11k_base *ab)
+{
+       ath11k_pci_ce_irq_disable_sync(ab);
+}
+
 static int ath11k_pci_map_service_to_pipe(struct ath11k_base *ab, u16 service_id,
                                          u8 *ul_pipe, u8 *dl_pipe)
 {
        .get_msi_address =  ath11k_pci_get_msi_address,
        .get_user_msi_vector = ath11k_get_user_msi_assignment,
        .map_service_to_pipe = ath11k_pci_map_service_to_pipe,
+       .ce_irq_enable = ath11k_pci_hif_ce_irq_enable,
+       .ce_irq_disable = ath11k_pci_hif_ce_irq_disable,
 };
 
 static int ath11k_pci_probe(struct pci_dev *pdev,