#include <linux/iommu.h>
 #include <linux/pci.h>
 #include <net/addrconf.h>
-#include <linux/idr.h>
 
 #include <linux/qed/qed_chain.h>
 #include <linux/qed/qed_if.h>
                break;
        case EVENT_TYPE_SRQ:
                srq_id = (u16)roce_handle64;
-               spin_lock_irqsave(&dev->srqidr.idr_lock, flags);
-               srq = idr_find(&dev->srqidr.idr, srq_id);
+               xa_lock_irqsave(&dev->srqs, flags);
+               srq = xa_load(&dev->srqs, srq_id);
                if (srq) {
                        ibsrq = &srq->ibsrq;
                        if (ibsrq->event_handler) {
                                  "SRQ event with NULL pointer ibsrq. Handle=%llx\n",
                                  roce_handle64);
                }
-               spin_unlock_irqrestore(&dev->srqidr.idr_lock, flags);
+               xa_unlock_irqrestore(&dev->srqs, flags);
                DP_NOTICE(dev, "SRQ event %d on handle %p\n", e_code, srq);
        default:
                break;
 
 #define __QEDR_H__
 
 #include <linux/pci.h>
-#include <linux/idr.h>
+#include <linux/xarray.h>
 #include <rdma/ib_addr.h>
 #include <linux/qed/qed_if.h>
 #include <linux/qed/qed_chain.h>
 
 #define QEDR_ENET_STATE_BIT    (0)
 
-struct qedr_idr {
-       spinlock_t idr_lock; /* Protect idr data-structure */
-       struct idr idr;
-};
-
 struct qedr_dev {
        struct ib_device        ibdev;
        struct qed_dev          *cdev;
        struct qedr_qp          *gsi_qp;
        enum qed_rdma_type      rdma_type;
        struct xarray           qps;
-       struct qedr_idr         srqidr;
+       struct xarray           srqs;
        struct workqueue_struct *iwarp_wq;
        u16                     iwarp_max_mtu;
 
 
        return rc;
 }
 
-static int qedr_idr_add(struct qedr_dev *dev, struct qedr_idr *qidr,
-                       void *ptr, u32 id);
-static void qedr_idr_remove(struct qedr_dev *dev,
-                           struct qedr_idr *qidr, u32 id);
-
 struct ib_srq *qedr_create_srq(struct ib_pd *ibpd,
                               struct ib_srq_init_attr *init_attr,
                               struct ib_udata *udata)
                        goto err2;
        }
 
-       rc = qedr_idr_add(dev, &dev->srqidr, srq, srq->srq_id);
+       rc = xa_insert_irq(&dev->srqs, srq->srq_id, srq, GFP_KERNEL);
        if (rc)
                goto err2;
 
        struct qedr_dev *dev = get_qedr_dev(ibsrq->device);
        struct qedr_srq *srq = get_qedr_srq(ibsrq);
 
-       qedr_idr_remove(dev, &dev->srqidr, srq->srq_id);
+       xa_erase_irq(&dev->srqs, srq->srq_id);
        in_params.srq_id = srq->srq_id;
        dev->ops->rdma_destroy_srq(dev->rdma_ctx, &in_params);
 
                 qp->usq.buf_len, qp->urq.buf_addr, qp->urq.buf_len);
 }
 
-static int qedr_idr_add(struct qedr_dev *dev, struct qedr_idr *qidr,
-                       void *ptr, u32 id)
-{
-       int rc;
-
-       idr_preload(GFP_KERNEL);
-       spin_lock_irq(&qidr->idr_lock);
-
-       rc = idr_alloc(&qidr->idr, ptr, id, id + 1, GFP_ATOMIC);
-
-       spin_unlock_irq(&qidr->idr_lock);
-       idr_preload_end();
-
-       return rc < 0 ? rc : 0;
-}
-
-static void qedr_idr_remove(struct qedr_dev *dev, struct qedr_idr *qidr, u32 id)
-{
-       spin_lock_irq(&qidr->idr_lock);
-       idr_remove(&qidr->idr, id);
-       spin_unlock_irq(&qidr->idr_lock);
-}
-
 static inline void
 qedr_iwarp_populate_user_qp(struct qedr_dev *dev,
                            struct qedr_qp *qp,