return 0;
 }
 
-int __init init_hw_perf_events(void)
+int init_hw_perf_events(void)
 {
        csky_pmu.pmu = (struct pmu) {
                .pmu_enable     = csky_pmu_enable,
 
        return perf_pmu_register(&csky_pmu.pmu, "cpu", PERF_TYPE_RAW);
 }
-arch_initcall(init_hw_perf_events);
+
+int csky_pmu_device_probe(struct platform_device *pdev,
+                         const struct of_device_id *of_table)
+{
+       int ret;
+
+       ret = init_hw_perf_events();
+       if (ret) {
+               pr_notice("[perf] failed to probe PMU!\n");
+               return ret;
+       }
+
+       return ret;
+}
+
+const static struct of_device_id csky_pmu_of_device_ids[] = {
+       {.compatible = "csky,csky-pmu"},
+       {},
+};
+
+static int csky_pmu_dev_probe(struct platform_device *pdev)
+{
+       return csky_pmu_device_probe(pdev, csky_pmu_of_device_ids);
+}
+
+static struct platform_driver csky_pmu_driver = {
+       .driver = {
+                  .name = "csky-pmu",
+                  .of_match_table = csky_pmu_of_device_ids,
+                  },
+       .probe = csky_pmu_dev_probe,
+};
+
+static int __init csky_pmu_probe(void)
+{
+       int ret;
+
+       ret = platform_driver_register(&csky_pmu_driver);
+       if (ret)
+               pr_notice("[perf] PMU initialization failed\n");
+       else
+               pr_notice("[perf] PMU initialization done\n");
+
+       return ret;
+}
+
+device_initcall(csky_pmu_probe);