iommu/mediatek: Fix M4Uv1 group refcounting

[mirror_ubuntu-zesty-kernel.git] / drivers / iommu / mtk_iommu_v1.c

diff --git a/drivers/iommu/mtk_iommu_v1.c b/drivers/iommu/mtk_iommu_v1.c
index b8aeb07684836c902a9f87829985f05a44fdcc5d..c7063e9d67d8ed882111266dcffe6aa46152c6cc 100644 (file)
--- a/drivers/iommu/mtk_iommu_v1.c
+++ b/drivers/iommu/mtk_iommu_v1.c
@@ -502,6 +502,8 @@ static struct iommu_group *mtk_iommu_device_group(struct device *dev)
                data->m4u_group = iommu_group_alloc();
                if (IS_ERR(data->m4u_group))
                        dev_err(dev, "Failed to allocate M4U IOMMU group\n");
+       } else {
+               iommu_group_ref_get(data->m4u_group);
        }
        return data->m4u_group;
 }