1 files changed, 2 insertions, 4 deletions

diff --git a/drivers/iommu/qcom_iommu.c b/drivers/iommu/qcom_iommu.c
index 65b9c99707f8..fe88a4880d3a 100644
--- a/drivers/iommu/qcom_iommu.c
+++ b/drivers/iommu/qcom_iommu.c
@@ -885,16 +885,14 @@ static int qcom_iommu_device_remove(struct platform_device *pdev)
 
 static int __maybe_unused qcom_iommu_resume(struct device *dev)
 {
-        struct platform_device *pdev = to_platform_device(dev);
-        struct qcom_iommu_dev *qcom_iommu = platform_get_drvdata(pdev);
+        struct qcom_iommu_dev *qcom_iommu = dev_get_drvdata(dev);
 
         return qcom_iommu_enable_clocks(qcom_iommu);
 }
 
 static int __maybe_unused qcom_iommu_suspend(struct device *dev)
 {
-        struct platform_device *pdev = to_platform_device(dev);
-        struct qcom_iommu_dev *qcom_iommu = platform_get_drvdata(pdev);
+        struct qcom_iommu_dev *qcom_iommu = dev_get_drvdata(dev);
 
         qcom_iommu_disable_clocks(qcom_iommu);