2 files changed, 10 insertions, 24 deletions

diff --git a/drivers/thermal/cpu_cooling.c b/drivers/thermal/cpu_cooling.c
index 836828e29a87..c33fa5315d6b 100644
--- a/drivers/thermal/cpu_cooling.c
+++ b/drivers/thermal/cpu_cooling.c
@@ -73,21 +73,14 @@ static struct cpufreq_cooling_device *notify_device;
  */
 static int get_idr(struct idr *idr, int *id)
 {
-        int err;
-again:
-        if (unlikely(idr_pre_get(idr, GFP_KERNEL) == 0))
-                return -ENOMEM;
+        int ret;
 
         mutex_lock(&cooling_cpufreq_lock);
-        err = idr_get_new(idr, NULL, id);
+        ret = idr_alloc(idr, NULL, 0, 0, GFP_KERNEL);
         mutex_unlock(&cooling_cpufreq_lock);
-
-        if (unlikely(err == -EAGAIN))
-                goto again;
-        else if (unlikely(err))
-                return err;
-
-        *id = *id & MAX_IDR_MASK;
+        if (unlikely(ret < 0))
+                return ret;
+        *id = ret;
         return 0;
 }
 
diff --git a/drivers/thermal/thermal_sys.c b/drivers/thermal/thermal_sys.c
index 8c8ce806180f..84e95f32cdb6 100644
--- a/drivers/thermal/thermal_sys.c
+++ b/drivers/thermal/thermal_sys.c
@@ -132,23 +132,16 @@ EXPORT_SYMBOL_GPL(thermal_unregister_governor);
 
 static int get_idr(struct idr *idr, struct mutex *lock, int *id)
 {
-        int err;
-
-again:
-        if (unlikely(idr_pre_get(idr, GFP_KERNEL) == 0))
-                return -ENOMEM;
+        int ret;
 
         if (lock)
                 mutex_lock(lock);
-        err = idr_get_new(idr, NULL, id);
+        ret = idr_alloc(idr, NULL, 0, 0, GFP_KERNEL);
         if (lock)
                 mutex_unlock(lock);
-        if (unlikely(err == -EAGAIN))
-                goto again;
-        else if (unlikely(err))
-                return err;
-
-        *id = *id & MAX_IDR_MASK;
+        if (unlikely(ret < 0))
+                return ret;
+        *id = ret;
         return 0;
 }