enum ice_disq_rst_src rst_src, u16 vmvf_num,
                struct ice_sq_cd *cd)
 {
-       struct ice_aqc_dis_txq_item *qg_list;
+       DEFINE_FLEX(struct ice_aqc_dis_txq_item, qg_list, q_id, 1);
+       u16 i, buf_size = __struct_size(qg_list);
        struct ice_q_ctx *q_ctx;
        int status = -ENOENT;
        struct ice_hw *hw;
-       u16 i, buf_size;
 
        if (!pi || pi->port_state != ICE_SCHED_PORT_STATE_READY)
                return -EIO;
                return -EIO;
        }
 
-       buf_size = struct_size(qg_list, q_id, 1);
-       qg_list = kzalloc(buf_size, GFP_KERNEL);
-       if (!qg_list)
-               return -ENOMEM;
-
        mutex_lock(&pi->sched_lock);
 
        for (i = 0; i < num_queues; i++) {
                q_ctx->q_teid = ICE_INVAL_TEID;
        }
        mutex_unlock(&pi->sched_lock);
-       kfree(qg_list);
        return status;
 }
 
 ice_dis_vsi_rdma_qset(struct ice_port_info *pi, u16 count, u32 *qset_teid,
                      u16 *q_id)
 {
-       struct ice_aqc_dis_txq_item *qg_list;
+       DEFINE_FLEX(struct ice_aqc_dis_txq_item, qg_list, q_id, 1);
+       u16 qg_size = __struct_size(qg_list);
        struct ice_hw *hw;
        int status = 0;
-       u16 qg_size;
        int i;
 
        if (!pi || pi->port_state != ICE_SCHED_PORT_STATE_READY)
 
        hw = pi->hw;
 
-       qg_size = struct_size(qg_list, q_id, 1);
-       qg_list = kzalloc(qg_size, GFP_KERNEL);
-       if (!qg_list)
-               return -ENOMEM;
-
        mutex_lock(&pi->sched_lock);
 
        for (i = 0; i < count; i++) {
        }
 
        mutex_unlock(&pi->sched_lock);
-       kfree(qg_list);
        return status;
 }