config ATH11K
        tristate "Qualcomm Technologies 802.11ax chipset support"
        depends on MAC80211 && HAS_DMA
-       depends on REMOTEPROC
        depends on CRYPTO_MICHAEL_MIC
        select ATH_COMMON
        select QCOM_QMI_HELPERS
 config ATH11K_AHB
        tristate "Atheros ath11k AHB support"
        depends on ATH11K
+       depends on REMOTEPROC
        help
          This module adds support for AHB bus
 
 
 
 MODULE_DEVICE_TABLE(of, ath11k_ahb_of_match);
 
+static const struct ath11k_bus_params ath11k_ahb_bus_params = {
+       .mhi_support = false,
+};
+
 /* Target firmware's Copy Engine configuration. */
 static const struct ce_pipe_config target_ce_config_wlan[] = {
        /* CE0: host->target HTC control and raw streams */
                return ret;
        }
 
-       ab = ath11k_core_alloc(&pdev->dev, 0, ATH11K_BUS_AHB);
+       ab = ath11k_core_alloc(&pdev->dev, 0, ATH11K_BUS_AHB, &ath11k_ahb_bus_params);
        if (!ab) {
                dev_err(&pdev->dev, "failed to allocate ath11k base\n");
                return -ENOMEM;
 
 }
 EXPORT_SYMBOL(ath11k_core_pre_init);
 
-int ath11k_core_init(struct ath11k_base *ab)
+static int ath11k_core_get_rproc(struct ath11k_base *ab)
 {
        struct device *dev = ab->dev;
        struct rproc *prproc;
        phandle rproc_phandle;
-       int ret;
+
+       if (ab->bus_params.mhi_support)
+               return 0;
 
        if (of_property_read_u32(dev->of_node, "qcom,rproc", &rproc_phandle)) {
                ath11k_err(ab, "failed to get q6_rproc handle\n");
        }
        ab->tgt_rproc = prproc;
 
+       return 0;
+}
+
+int ath11k_core_init(struct ath11k_base *ab)
+{
+       int ret;
+
+       ret = ath11k_core_get_rproc(ab);
+       if (ret) {
+               ath11k_err(ab, "failed to get rproc: %d\n", ret);
+               return ret;
+       }
+
+       ret = ath11k_init_hw_params(ab);
+       if (ret) {
+               ath11k_err(ab, "failed to get hw params %d\n", ret);
+               return ret;
+       }
+
        ret = ath11k_core_soc_create(ab);
        if (ret) {
                ath11k_err(ab, "failed to create soc core: %d\n", ret);
 EXPORT_SYMBOL(ath11k_core_free);
 
 struct ath11k_base *ath11k_core_alloc(struct device *dev, size_t priv_size,
-                                     enum ath11k_bus bus)
+                                     enum ath11k_bus bus,
+                                     const struct ath11k_bus_params *bus_params)
 {
        struct ath11k_base *ab;
 
        INIT_WORK(&ab->restart_work, ath11k_core_restart);
        timer_setup(&ab->rx_replenish_retry, ath11k_ce_rx_replenish_retry, 0);
        ab->dev = dev;
+       ab->bus_params = *bus_params;
 
        return ab;
 
 
        size_t len;
 };
 
+struct ath11k_bus_params {
+       bool mhi_support;
+};
+
 /* IPQ8074 HW channel counters frequency value in hertz */
 #define IPQ8074_CC_FREQ_HERTZ 320000
 
        u32 ext_service_bitmap[WMI_SERVICE_EXT_BM_SIZE];
        bool pdevs_macaddr_valid;
        int bd_api;
+
        struct ath11k_hw_params hw_params;
+       struct ath11k_bus_params bus_params;
+
        const struct firmware *cal_file;
 
        /* Below regd's are protected by ab->data_lock */
 int ath11k_core_init(struct ath11k_base *ath11k);
 void ath11k_core_deinit(struct ath11k_base *ath11k);
 struct ath11k_base *ath11k_core_alloc(struct device *dev, size_t priv_size,
-                                     enum ath11k_bus bus);
+                                     enum ath11k_bus bus,
+                                     const struct ath11k_bus_params *bus_params);
 void ath11k_core_free(struct ath11k_base *ath11k);
 int ath11k_core_fetch_bdf(struct ath11k_base *ath11k,
                          struct ath11k_board_data *bd);
 
 
 MODULE_DEVICE_TABLE(pci, ath11k_pci_id_table);
 
+static const struct ath11k_bus_params ath11k_pci_bus_params = {
+       .mhi_support = true,
+};
+
 static const struct ath11k_msi_config msi_config = {
        .total_vectors = 32,
        .total_users = 4,
                return -ENOTSUPP;
        }
 
-       ab = ath11k_core_alloc(&pdev->dev, sizeof(*ab_pci), ATH11K_BUS_PCI);
+       ab = ath11k_core_alloc(&pdev->dev, sizeof(*ab_pci), ATH11K_BUS_PCI,
+                              &ath11k_pci_bus_params);
        if (!ab) {
                dev_err(&pdev->dev, "failed to allocate ath11k base\n");
                return -ENOMEM;