dmaengine: idxd: use ida for device instance enumeration

[ Upstream commit f7f7739847bd68b3c3103fd1b50d943038bd14c7 ]

The idr is only used for an device id, never to lookup context from that
id. Switch to plain ida.

Fixes: bfe1d56091c1 ("dmaengine: idxd: Init and probe for Intel data accelerators")
Reported-by: Jason Gunthorpe <jgg@nvidia.com>
Signed-off-by: Dave Jiang <dave.jiang@intel.com>
Link: https://lore.kernel.org/r/161852984730.2203940.15032482460902003819.stgit@djiang5-desk3.ch.intel.com
Signed-off-by: Vinod Koul <vkoul@kernel.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/drivers/dma/idxd/init.c b/drivers/dma/idxd/init.c
index d223242a34f488e9b35ca16a56637f90e392695c..6e97a9870ba8db53faa74c9bdf32bd4991fcc123 100644 (file)
--- a/drivers/dma/idxd/init.c
+++ b/drivers/dma/idxd/init.c
@@ -30,8 +30,7 @@ MODULE_AUTHOR("Intel Corporation");
 
 bool support_enqcmd;
 
-static struct idr idxd_idrs[IDXD_TYPE_MAX];
-static DEFINE_MUTEX(idxd_idr_lock);
+static struct ida idxd_idas[IDXD_TYPE_MAX];
 
 static struct pci_device_id idxd_pci_tbl[] = {
        /* DSA ver 1.0 platforms */
@@ -342,12 +341,10 @@ static int idxd_probe(struct idxd_device *idxd)
 
        dev_dbg(dev, "IDXD interrupt setup complete.\n");
 
-       mutex_lock(&idxd_idr_lock);
-       idxd->id = idr_alloc(&idxd_idrs[idxd->type], idxd, 0, 0, GFP_KERNEL);
-       mutex_unlock(&idxd_idr_lock);
+       idxd->id = ida_alloc(&idxd_idas[idxd->type], GFP_KERNEL);
        if (idxd->id < 0) {
                rc = -ENOMEM;
-               goto err_idr_fail;
+               goto err_ida_fail;
        }
 
        idxd->major = idxd_cdev_get_major(idxd);
@@ -355,7 +352,7 @@ static int idxd_probe(struct idxd_device *idxd)
        dev_dbg(dev, "IDXD device %d probed successfully\n", idxd->id);
        return 0;
 
- err_idr_fail:
+ err_ida_fail:
        idxd_mask_error_interrupts(idxd);
        idxd_mask_msix_vectors(idxd);
  err_setup:
@@ -512,9 +509,7 @@ static void idxd_remove(struct pci_dev *pdev)
        idxd_shutdown(pdev);
        if (device_pasid_enabled(idxd))
                idxd_disable_system_pasid(idxd);
-       mutex_lock(&idxd_idr_lock);
-       idr_remove(&idxd_idrs[idxd->type], idxd->id);
-       mutex_unlock(&idxd_idr_lock);
+       ida_free(&idxd_idas[idxd->type], idxd->id);
 }
 
 static struct pci_driver idxd_pci_driver = {
@@ -544,7 +539,7 @@ static int __init idxd_init_module(void)
                support_enqcmd = true;
 
        for (i = 0; i < IDXD_TYPE_MAX; i++)
-               idr_init(&idxd_idrs[i]);
+               ida_init(&idxd_idas[i]);
 
        err = idxd_register_bus_type();
        if (err < 0)