return is_cmd_complete;
 }
 
+/**
+ *****************************************************************************
+ *  Function: dc_process_dmub_set_mst_slots
+ *
+ *  @brief
+ *             Submits mst slot allocation command to dmub via inbox message
+ *
+ *  @param
+ *             [in] dc: dc structure
+ *             [in] link_index: link index
+ *             [in] mst_alloc_slots: mst slots to be allotted
+ *             [out] mst_slots_in_use: mst slots in use returned in failure case
+ *
+ *     @return
+ *             DC_OK if successful, DC_ERROR if failure
+ *****************************************************************************
+ */
+enum dc_status dc_process_dmub_set_mst_slots(const struct dc *dc,
+                               uint32_t link_index,
+                               uint8_t mst_alloc_slots,
+                               uint8_t *mst_slots_in_use)
+{
+       union dmub_rb_cmd cmd = {0};
+       struct dc_dmub_srv *dmub_srv = dc->ctx->dmub_srv;
+
+       /* prepare MST_ALLOC_SLOTS command */
+       cmd.set_mst_alloc_slots.header.type = DMUB_CMD__DPIA;
+       cmd.set_mst_alloc_slots.header.sub_type = DMUB_CMD__DPIA_MST_ALLOC_SLOTS;
+
+       cmd.set_mst_alloc_slots.mst_slots_control.instance = dc->links[link_index]->ddc_hw_inst;
+       cmd.set_mst_alloc_slots.mst_slots_control.mst_alloc_slots = mst_alloc_slots;
+
+       if (!dc_dmub_srv_cmd_with_reply_data(dmub_srv, &cmd))
+               /* command is not processed by dmub */
+               return DC_ERROR_UNEXPECTED;
+
+       /* command processed by dmub, if ret_status is 1 */
+       if (cmd.set_config_access.header.ret_status != 1)
+               /* command processing error */
+               return DC_ERROR_UNEXPECTED;
+
+       /* command processed and we have a status of 2, mst not enabled in dpia */
+       if (cmd.set_mst_alloc_slots.mst_slots_control.immed_status == 2)
+               return DC_FAIL_UNSUPPORTED_1;
+
+       /* previously configured mst alloc and used slots did not match */
+       if (cmd.set_mst_alloc_slots.mst_slots_control.immed_status == 3) {
+               *mst_slots_in_use = cmd.set_mst_alloc_slots.mst_slots_control.mst_slots_in_use;
+               return DC_NOT_SUPPORTED;
+       }
+
+       return DC_OK;
+}
+
 /**
  * dc_disable_accelerated_mode - disable accelerated mode
  * @dc: dc structure
 
 
        ASSERT(proposed_table.stream_count > 0);
 
+       if (link->ep_type == DISPLAY_ENDPOINT_USB4_DPIA) {
+               static enum dc_status status;
+               uint8_t mst_alloc_slots = 0, prev_mst_slots_in_use = 0xFF;
+
+               for (i = 0; i < link->mst_stream_alloc_table.stream_count; i++)
+                       mst_alloc_slots += link->mst_stream_alloc_table.stream_allocations[i].slot_count;
+
+               status = dc_process_dmub_set_mst_slots(link->dc, link->link_index,
+                       mst_alloc_slots, &prev_mst_slots_in_use);
+               ASSERT(status == DC_OK);
+               DC_LOG_MST("dpia : status[%d]: alloc_slots[%d]: used_slots[%d]\n",
+                               status, mst_alloc_slots, prev_mst_slots_in_use);
+       }
+
        /* program DP source TX for payload */
 #if defined(CONFIG_DRM_AMD_DC_DCN)
        switch (dp_get_link_encoding_format(&link->cur_link_settings)) {
 #endif
        }
 
+       if (link->ep_type == DISPLAY_ENDPOINT_USB4_DPIA) {
+               enum dc_status status;
+               uint8_t mst_alloc_slots = 0, prev_mst_slots_in_use = 0xFF;
+
+               for (i = 0; i < link->mst_stream_alloc_table.stream_count; i++)
+                       mst_alloc_slots += link->mst_stream_alloc_table.stream_allocations[i].slot_count;
+
+               status = dc_process_dmub_set_mst_slots(link->dc, link->link_index,
+                       mst_alloc_slots, &prev_mst_slots_in_use);
+               ASSERT(status != DC_NOT_SUPPORTED);
+               DC_LOG_MST("dpia : status[%d]: alloc_slots[%d]: used_slots[%d]\n",
+                               status, mst_alloc_slots, prev_mst_slots_in_use);
+       }
+
 #if defined(CONFIG_DRM_AMD_DC_DCN)
        switch (dp_get_link_encoding_format(&link->cur_link_settings)) {
        case DP_8b_10b_ENCODING:
 
                                uint32_t link_index,
                                struct set_config_cmd_payload *payload,
                                struct dmub_notification *notify);
+
+enum dc_status dc_process_dmub_set_mst_slots(const struct dc *dc,
+                               uint32_t link_index,
+                               uint8_t mst_alloc_slots,
+                               uint8_t *mst_slots_in_use);
+
 /*******************************************************************************
  * DSC Interfaces
  ******************************************************************************/
 
 enum dmub_cmd_dpia_type {
        DMUB_CMD__DPIA_DIG1_DPIA_CONTROL = 0,
        DMUB_CMD__DPIA_SET_CONFIG_ACCESS = 1,
+       DMUB_CMD__DPIA_MST_ALLOC_SLOTS = 2,
 };
 
 #pragma pack(push, 1)
        struct dmub_cmd_set_config_control_data set_config_control; /* set config data */
 };
 
+/**
+ * Data passed from driver to FW in a DMUB_CMD__DPIA_MST_ALLOC_SLOTS command.
+ */
+struct dmub_cmd_mst_alloc_slots_control_data {
+       uint8_t mst_alloc_slots; /* mst slots to be allotted */
+       uint8_t instance; /* DPIA instance */
+       uint8_t immed_status; /* Immediate status returned as there is no outbox msg posted */
+       uint8_t mst_slots_in_use; /* returns slots in use for error cases */
+};
+
+/**
+ * DMUB command structure for SET_ command.
+ */
+struct dmub_rb_cmd_set_mst_alloc_slots {
+       struct dmub_cmd_header header; /* header */
+       struct dmub_cmd_mst_alloc_slots_control_data mst_slots_control; /* mst slots control */
+};
+
 /**
  * struct dmub_rb_cmd_dpphy_init - DPPHY init.
  */
         * Definition of a DMUB_CMD__DPIA_SET_CONFIG_ACCESS command.
         */
        struct dmub_rb_cmd_set_config_access set_config_access;
+       /**
+        * Definition of a DMUB_CMD__DPIA_MST_ALLOC_SLOTS command.
+        */
+       struct dmub_rb_cmd_set_mst_alloc_slots set_mst_alloc_slots;
        /**
         * Definition of a DMUB_CMD__EDID_CEA command.
         */