return ret;
 }
 
-static int qcom_tbu_probe(struct platform_device *pdev)
+int qcom_tbu_probe(struct platform_device *pdev)
 {
        struct of_phandle_args args = { .args_count = 2 };
        struct device_node *np = pdev->dev.of_node;
 
        return 0;
 }
-
-static const struct of_device_id qcom_tbu_of_match[] = {
-       { .compatible = "qcom,sc7280-tbu" },
-       { .compatible = "qcom,sdm845-tbu" },
-       { }
-};
-
-static struct platform_driver qcom_tbu_driver = {
-       .driver = {
-               .name           = "qcom_tbu",
-               .of_match_table = qcom_tbu_of_match,
-       },
-       .probe = qcom_tbu_probe,
-};
-builtin_platform_driver(qcom_tbu_driver);
 
 #include <linux/delay.h>
 #include <linux/of_device.h>
 #include <linux/firmware/qcom/qcom_scm.h>
+#include <linux/platform_device.h>
+#include <linux/pm_runtime.h>
 
 #include "arm-smmu.h"
 #include "arm-smmu-qcom.h"
 };
 #endif
 
+static int qcom_smmu_tbu_probe(struct platform_device *pdev)
+{
+       struct device *dev = &pdev->dev;
+       int ret;
+
+       if (IS_ENABLED(CONFIG_ARM_SMMU_QCOM_DEBUG)) {
+               ret = qcom_tbu_probe(pdev);
+               if (ret)
+                       return ret;
+       }
+
+       if (dev->pm_domain) {
+               pm_runtime_set_active(dev);
+               pm_runtime_enable(dev);
+       }
+
+       return 0;
+}
+
+static const struct of_device_id qcom_smmu_tbu_of_match[] = {
+       { .compatible = "qcom,sc7280-tbu" },
+       { .compatible = "qcom,sdm845-tbu" },
+       { }
+};
+
+static struct platform_driver qcom_smmu_tbu_driver = {
+       .driver = {
+               .name           = "qcom_tbu",
+               .of_match_table = qcom_smmu_tbu_of_match,
+       },
+       .probe = qcom_smmu_tbu_probe,
+};
+
 struct arm_smmu_device *qcom_smmu_impl_init(struct arm_smmu_device *smmu)
 {
        const struct device_node *np = smmu->dev->of_node;
        const struct of_device_id *match;
+       static u8 tbu_registered;
+
+       if (!tbu_registered++)
+               platform_driver_register(&qcom_smmu_tbu_driver);
 
 #ifdef CONFIG_ACPI
        if (np == NULL) {
 
 
 #ifdef CONFIG_ARM_SMMU_QCOM_DEBUG
 void qcom_smmu_tlb_sync_debug(struct arm_smmu_device *smmu);
+int qcom_tbu_probe(struct platform_device *pdev);
 #else
 static inline void qcom_smmu_tlb_sync_debug(struct arm_smmu_device *smmu) { }
+static inline int qcom_tbu_probe(struct platform_device *pdev) { return -EINVAL; }
 #endif
 
 #endif /* _ARM_SMMU_QCOM_H */