1 files changed, 3 insertions, 7 deletions

diff --git a/drivers/remoteproc/remoteproc_core.c b/drivers/remoteproc/remoteproc_core.c
index 634d36793b72..29387df4bfc9 100644
--- a/drivers/remoteproc/remoteproc_core.c
+++ b/drivers/remoteproc/remoteproc_core.c
@@ -199,11 +199,6 @@ int rproc_alloc_vring(struct rproc_vdev *rvdev, int i)
         /* actual size of vring (in bytes) */
         size = PAGE_ALIGN(vring_size(rvring->len, rvring->align));
 
-        if (!idr_pre_get(&rproc->notifyids, GFP_KERNEL)) {
-                dev_err(dev, "idr_pre_get failed\n");
-                return -ENOMEM;
-        }
-
         /*
          * Allocate non-cacheable memory for the vring. In the future
          * this call will also configure the IOMMU for us
@@ -221,12 +216,13 @@ int rproc_alloc_vring(struct rproc_vdev *rvdev, int i)
          * TODO: let the rproc know the notifyid of this vring
          * TODO: support predefined notifyids (via resource table)
          */
-        ret = idr_get_new(&rproc->notifyids, rvring, &notifyid);
+        ret = idr_alloc(&rproc->notifyids, rvring, 0, 0, GFP_KERNEL);
         if (ret) {
-                dev_err(dev, "idr_get_new failed: %d\n", ret);
+                dev_err(dev, "idr_alloc failed: %d\n", ret);
                 dma_free_coherent(dev->parent, size, va, dma);
                 return ret;
         }
+        notifyid = ret;
 
         /* Store largest notifyid */
         rproc->max_notifyid = max(rproc->max_notifyid, notifyid);