*     5 or less       sw_entry_point_layer
         */
        /* calculate the VSI layer based on number of layers. */
-       if (hw->num_tx_sched_layers > ICE_VSI_LAYER_OFFSET + 1) {
-               u8 layer = hw->num_tx_sched_layers - ICE_VSI_LAYER_OFFSET;
-
-               if (layer > hw->sw_entry_point_layer)
-                       return layer;
-       }
+       if (hw->num_tx_sched_layers == ICE_SCHED_9_LAYERS)
+               return hw->num_tx_sched_layers - ICE_VSI_LAYER_OFFSET;
+       else if (hw->num_tx_sched_layers == ICE_SCHED_5_LAYERS)
+               /* qgroup and VSI layers are same */
+               return hw->num_tx_sched_layers - ICE_QGRP_LAYER_OFFSET;
        return hw->sw_entry_point_layer;
 }
 
         *     7 or less       sw_entry_point_layer
         */
        /* calculate the aggregator layer based on number of layers. */
-       if (hw->num_tx_sched_layers > ICE_AGG_LAYER_OFFSET + 1) {
-               u8 layer = hw->num_tx_sched_layers - ICE_AGG_LAYER_OFFSET;
-
-               if (layer > hw->sw_entry_point_layer)
-                       return layer;
-       }
-       return hw->sw_entry_point_layer;
+       if (hw->num_tx_sched_layers == ICE_SCHED_9_LAYERS)
+               return hw->num_tx_sched_layers - ICE_AGG_LAYER_OFFSET;
+       else
+               return hw->sw_entry_point_layer;
 }
 
 /**
 {
        struct ice_sched_node *vsi_node, *qgrp_node;
        struct ice_vsi_ctx *vsi_ctx;
+       u8 qgrp_layer, vsi_layer;
        u16 max_children;
-       u8 qgrp_layer;
 
        qgrp_layer = ice_sched_get_qgrp_layer(pi->hw);
+       vsi_layer = ice_sched_get_vsi_layer(pi->hw);
        max_children = pi->hw->max_children[qgrp_layer];
 
        vsi_ctx = ice_get_vsi_ctx(pi->hw, vsi_handle);
        if (!vsi_node)
                return NULL;
 
+       /* If the queue group and VSI layer are same then queues
+        * are all attached directly to VSI
+        */
+       if (qgrp_layer == vsi_layer)
+               return vsi_node;
+
        /* get the first queue group node from VSI sub-tree */
        qgrp_node = ice_sched_get_first_node(pi, vsi_node, qgrp_layer);
        while (qgrp_node) {
        u8 profile_type;
        int status;
 
-       if (layer_num >= ICE_AQC_TOPO_MAX_LEVEL_NUM)
+       if (!pi || layer_num >= pi->hw->num_tx_sched_layers)
                return NULL;
        switch (rl_type) {
        case ICE_MIN_BW:
                return NULL;
        }
 
-       if (!pi)
-               return NULL;
        hw = pi->hw;
        list_for_each_entry(rl_prof_elem, &pi->rl_prof_list[layer_num],
                            list_entry)
        struct ice_aqc_rl_profile_info *rl_prof_elem;
        int status = 0;
 
-       if (layer_num >= ICE_AQC_TOPO_MAX_LEVEL_NUM)
+       if (layer_num >= pi->hw->num_tx_sched_layers)
                return -EINVAL;
        /* Check the existing list for RL profile */
        list_for_each_entry(rl_prof_elem, &pi->rl_prof_list[layer_num],