attr->local_ca_ack_delay = qattr->dev_ack_delay;
        attr->max_fast_reg_page_list_len = qattr->max_mr / 8;
-       attr->max_pkeys = QEDR_ROCE_PKEY_MAX;
+       attr->max_pkeys = qattr->max_pkey;
        attr->max_ah = qattr->max_ah;
 
        return 0;
 
        dev->max_mw = 0;
        dev->max_mr_mw_fmr_pbl = (PAGE_SIZE / 8) * (PAGE_SIZE / 8);
        dev->max_mr_mw_fmr_size = dev->max_mr_mw_fmr_pbl * PAGE_SIZE;
-       dev->max_pkey = QED_RDMA_MAX_P_KEY;
+       if (QED_IS_ROCE_PERSONALITY(p_hwfn))
+               dev->max_pkey = QED_RDMA_MAX_P_KEY;
 
        dev->max_srq = p_hwfn->p_rdma_info->num_srqs;
        dev->max_srq_wr = QED_RDMA_MAX_SRQ_WQE_ELEM;