return ret;
 }
 
-static void __init etm4_pm_clear(void)
+static void etm4_pm_clear(void)
 {
        cpu_pm_unregister_notifier(&etm4_cpu_pm_nb);
        cpuhp_remove_state_nocalls(CPUHP_AP_ARM_CORESIGHT_STARTING);
        if (!desc.name)
                return -ENOMEM;
 
-       etmdrvdata[drvdata->cpu] = drvdata;
-
        if (smp_call_function_single(drvdata->cpu,
                                etm4_init_arch_data,  drvdata, 1))
                dev_err(dev, "ETM arch init failed\n");
 
-       if (etm4_arch_supported(drvdata->arch) == false) {
-               ret = -EINVAL;
-               goto err_arch_supported;
-       }
+       if (etm4_arch_supported(drvdata->arch) == false)
+               return -EINVAL;
 
        etm4_init_trace_id(drvdata);
        etm4_set_default(&drvdata->config);
 
        pdata = coresight_get_platform_data(dev);
-       if (IS_ERR(pdata)) {
-               ret = PTR_ERR(pdata);
-               goto err_arch_supported;
-       }
+       if (IS_ERR(pdata))
+               return PTR_ERR(pdata);
+
        adev->dev.platform_data = pdata;
 
        desc.type = CORESIGHT_DEV_TYPE_SOURCE;
        desc.dev = dev;
        desc.groups = coresight_etmv4_groups;
        drvdata->csdev = coresight_register(&desc);
-       if (IS_ERR(drvdata->csdev)) {
-               ret = PTR_ERR(drvdata->csdev);
-               goto err_arch_supported;
-       }
+       if (IS_ERR(drvdata->csdev))
+               return PTR_ERR(drvdata->csdev);
 
        ret = etm_perf_symlink(drvdata->csdev, true);
        if (ret) {
                coresight_unregister(drvdata->csdev);
-               goto err_arch_supported;
+               return ret;
        }
 
+       etmdrvdata[drvdata->cpu] = drvdata;
+
        pm_runtime_put(&adev->dev);
        dev_info(&drvdata->csdev->dev, "CPU%d: ETM v%d.%d initialized\n",
                 drvdata->cpu, drvdata->arch >> 4, drvdata->arch & 0xf);
        }
 
        return 0;
-
-err_arch_supported:
-       etmdrvdata[drvdata->cpu] = NULL;
-       return ret;
 }
 
 static struct amba_cs_uci_id uci_id_etm4[] = {
        }
 };
 
+static void __exit clear_etmdrvdata(void *info)
+{
+       int cpu = *(int *)info;
+
+       etmdrvdata[cpu] = NULL;
+}
+
+static int __exit etm4_remove(struct amba_device *adev)
+{
+       struct etmv4_drvdata *drvdata = dev_get_drvdata(&adev->dev);
+
+       etm_perf_symlink(drvdata->csdev, false);
+
+       /*
+        * Taking hotplug lock here to avoid racing between etm4_remove and
+        * CPU hotplug call backs.
+        */
+       cpus_read_lock();
+       /*
+        * The readers for etmdrvdata[] are CPU hotplug call backs
+        * and PM notification call backs. Change etmdrvdata[i] on
+        * CPU i ensures these call backs has consistent view
+        * inside one call back function.
+        */
+       if (smp_call_function_single(drvdata->cpu, clear_etmdrvdata, &drvdata->cpu, 1))
+               etmdrvdata[drvdata->cpu] = NULL;
+
+       cpus_read_unlock();
+
+       coresight_unregister(drvdata->csdev);
+
+       return 0;
+}
+
 static const struct amba_id etm4_ids[] = {
        CS_AMBA_ID(0x000bb95d),                 /* Cortex-A53 */
        CS_AMBA_ID(0x000bb95e),                 /* Cortex-A57 */
        {},
 };
 
+MODULE_DEVICE_TABLE(amba, etm4_ids);
+
 static struct amba_driver etm4x_driver = {
        .drv = {
                .name   = "coresight-etm4x",
+               .owner  = THIS_MODULE,
                .suppress_bind_attrs = true,
        },
        .probe          = etm4_probe,
+       .remove         = etm4_remove,
        .id_table       = etm4_ids,
 };
 
 
        return ret;
 }
-device_initcall(etm4x_init);
+
+static void __exit etm4x_exit(void)
+{
+       amba_driver_unregister(&etm4x_driver);
+       etm4_pm_clear();
+}
+
+module_init(etm4x_init);
+module_exit(etm4x_exit);
+
+MODULE_AUTHOR("Pratik Patel <pratikp@codeaurora.org>");
+MODULE_AUTHOR("Mathieu Poirier <mathieu.poirier@linaro.org>");
+MODULE_DESCRIPTION("Arm CoreSight Program Flow Trace v4.x driver");
+MODULE_LICENSE("GPL v2");