#include <linux/of_address.h>
 #include <linux/of_device.h>
 #include <linux/perf_event.h>
+#include <linux/perf/arm_pmu.h>
 #include <linux/platform_device.h>
 #include <linux/printk.h>
 #include <linux/slab.h>
 };
 MODULE_DEVICE_TABLE(of, arm_spe_pmu_of_match);
 
-static int arm_spe_pmu_device_dt_probe(struct platform_device *pdev)
+static const struct platform_device_id arm_spe_match[] = {
+       { ARMV8_SPE_PDEV_NAME, 0},
+       { }
+};
+MODULE_DEVICE_TABLE(platform, arm_spe_match);
+
+static int arm_spe_pmu_device_probe(struct platform_device *pdev)
 {
        int ret;
        struct arm_spe_pmu *spe_pmu;
 }
 
 static struct platform_driver arm_spe_pmu_driver = {
+       .id_table = arm_spe_match,
        .driver = {
                .name           = DRVNAME,
                .of_match_table = of_match_ptr(arm_spe_pmu_of_match),
        },
-       .probe  = arm_spe_pmu_device_dt_probe,
+       .probe  = arm_spe_pmu_device_probe,
        .remove = arm_spe_pmu_device_remove,
 };