qcacld-3.0: Populate mlo IE in assoc response only for mlo connection

Populate mlo IE in assoc response only for mlo connection.
Mlo IE in assoc response should be the last IE to pupulate.

Change-Id: Ibcf63159bde942d6e08eb544c1a799203f0360f3
CRs-Fixed: 3033878

core/mac/src/pe/lim/lim_send_management_frames.c

@@ -1587,11 +1587,6 @@ lim_send_assoc_rsp_mgmt_frame(struct mac_context *mac_ctx,
 						      &frm.eht_op);
 		}
 
-#ifdef WLAN_FEATURE_11BE_MLO
-		populate_dot11f_assoc_rsp_mlo_ie(mac_ctx, pe_session,
-						 sta, &frm);
- #endif
-
 		if (status_code == STATUS_ASSOC_REJECTED_TEMPORARILY) {
 			max_retries =
 			mac_ctx->mlme_cfg->gen.pmf_sa_query_max_retries;
@@ -1685,6 +1680,14 @@ lim_send_assoc_rsp_mgmt_frame(struct mac_context *mac_ctx,
 		lim_merge_extcap_struct(&(frm.ExtCap), &extracted_ext_cap,
 					true);
 
+	if (sta && lim_is_sta_eht_capable(sta) &&
+	    lim_is_session_eht_capable(pe_session) &&
+	    wlan_vdev_mlme_is_mlo_ap(pe_session->vdev)) {
+		pe_debug("Populate mlo IEs");
+		populate_dot11f_assoc_rsp_mlo_ie(mac_ctx, pe_session,
+						 sta, &frm);
+	}
+
 	/* Allocate a buffer for this frame: */
 	status = dot11f_get_packed_assoc_response_size(mac_ctx, &frm, &payload);
 	if (DOT11F_FAILED(status)) {