#include "rx_desc.h"
 
 struct ath12k_base;
+#define HAL_CE_REMAP_REG_BASE  (ab->ce_remap_base_addr)
 
 #define HAL_LINK_DESC_SIZE                     (32 << 2)
 #define HAL_LINK_DESC_ALIGN                    128
  * ath12k_hal_rx_desc_get_err().
  */
 
+#define HAL_IPQ5332_CE_WFSS_REG_BASE   0x740000
+#define HAL_IPQ5332_CE_SIZE            0x100000
+
 enum hal_srng_ring_id {
        HAL_SRNG_RING_ID_REO2SW0 = 0,
        HAL_SRNG_RING_ID_REO2SW1,
 
        .hal_ppe_rel_ring_base = 0x0000046c,
 
        /* CE address */
-       .hal_umac_ce0_src_reg_base = 0x00740000,
-       .hal_umac_ce0_dest_reg_base = 0x00741000,
-       .hal_umac_ce1_src_reg_base = 0x00742000,
-       .hal_umac_ce1_dest_reg_base = 0x00743000,
+       .hal_umac_ce0_src_reg_base = 0x00740000 -
+               HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .hal_umac_ce0_dest_reg_base = 0x00741000 -
+               HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .hal_umac_ce1_src_reg_base = 0x00742000 -
+               HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .hal_umac_ce1_dest_reg_base = 0x00743000 -
+               HAL_IPQ5332_CE_WFSS_REG_BASE,
 };
 
 static const struct ath12k_hw_regs wcn7850_regs = {
                            HAL_WBM_SW_COOKIE_CONV_CFG_WBM2SW4_EN,
 };
 
+static const struct ce_ie_addr ath12k_ce_ie_addr_ipq5332 = {
+       .ie1_reg_addr = CE_HOST_IE_ADDRESS - HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .ie2_reg_addr = CE_HOST_IE_2_ADDRESS - HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .ie3_reg_addr = CE_HOST_IE_3_ADDRESS - HAL_IPQ5332_CE_WFSS_REG_BASE,
+};
+
+static const struct ce_remap ath12k_ce_remap_ipq5332 = {
+       .base = HAL_IPQ5332_CE_WFSS_REG_BASE,
+       .size = HAL_IPQ5332_CE_SIZE,
+};
+
 static const struct ath12k_hw_params ath12k_hw_params[] = {
        {
                .name = "qcn9274 hw1.0",
                .iova_mask = 0,
 
                .supports_aspm = false,
+
+               .ce_ie_addr = NULL,
+               .ce_remap = NULL,
        },
        {
                .name = "wcn7850 hw2.0",
                .iova_mask = ATH12K_PCIE_MAX_PAYLOAD_SIZE - 1,
 
                .supports_aspm = true,
+
+               .ce_ie_addr = NULL,
+               .ce_remap = NULL,
        },
        {
                .name = "qcn9274 hw2.0",
                .iova_mask = 0,
 
                .supports_aspm = false,
+
+               .ce_ie_addr = NULL,
+               .ce_remap = NULL,
        },
        {
                .name = "ipq5332 hw1.0",
                .supports_dynamic_smps_6ghz = false,
                .iova_mask = 0,
                .supports_aspm = false,
+
+               .ce_ie_addr = &ath12k_ce_ie_addr_ipq5332,
+               .ce_remap = &ath12k_ce_remap_ipq5332,
        },
 };