Add a atomic_t variable for initialization of hw dev.
Signed-off-by: kyrie wu <kyrie.wu@mediatek.com>
Signed-off-by: irui wang <irui.wang@mediatek.com>
Signed-off-by: Hans Verkuil <hverkuil-cisco@xs4all.nl>
Signed-off-by: Mauro Carvalho Chehab <mchehab@kernel.org>
                        num_child++;
 
                atomic_set(&jpeg->hw_rdy, num_child);
+               atomic_set(&jpeg->hw_index, 0);
 
                jpeg->workqueue = alloc_ordered_workqueue(MTK_JPEG_NAME,
                                                          WQ_MEM_RECLAIM
 
  * @hw_rdy:            jpg hw ready flag
  * @reg_decbase:       jpg decode register base addr
  * @dec_hw_dev:        jpg decode hardware device
+ * @hw_index:          jpg hw index
  */
 struct mtk_jpeg_dev {
        struct mutex            lock;
 
        void __iomem *reg_decbase[MTK_JPEGDEC_HW_MAX];
        struct mtk_jpegdec_comp_dev *dec_hw_dev[MTK_JPEGDEC_HW_MAX];
+       atomic_t hw_index;
 };
 
 /**
 
                return dev_err_probe(&pdev->dev, ret,
                                     "Failed to register JPEGDEC irq handler.\n");
 
-       for (i = 0; i < MTK_JPEGDEC_HW_MAX; i++) {
-               if (master_dev->dec_hw_dev[i])
-                       continue;
-
-               master_dev->dec_hw_dev[i] = dev;
-               master_dev->reg_decbase[i] = dev->reg_base;
-               dev->master_dev = master_dev;
-       }
+       i = atomic_add_return(1, &master_dev->hw_index) - 1;
+       master_dev->dec_hw_dev[i] = dev;
+       master_dev->reg_decbase[i] = dev->reg_base;
+       dev->master_dev = master_dev;
 
        platform_set_drvdata(pdev, dev);
        pm_runtime_enable(&pdev->dev);
 
        if (ret)
                return ret;
 
-       for (i = 0; i < MTK_JPEGENC_HW_MAX; i++) {
-               if (master_dev->enc_hw_dev[i])
-                       continue;
-
-               master_dev->enc_hw_dev[i] = dev;
-               master_dev->reg_encbase[i] = dev->reg_base;
-               dev->master_dev = master_dev;
-       }
+       i = atomic_add_return(1, &master_dev->hw_index) - 1;
+       master_dev->enc_hw_dev[i] = dev;
+       master_dev->reg_encbase[i] = dev->reg_base;
+       dev->master_dev = master_dev;
 
        platform_set_drvdata(pdev, dev);
        pm_runtime_enable(&pdev->dev);