#include <linux/of_address.h>
 #include <linux/of_irq.h>
 #include <linux/platform_device.h>
-#include <linux/syscore_ops.h>
+#include <linux/pm.h>
 
 #include <dt-bindings/interrupt-controller/arm-gic.h>
 
        bool dt_has_irqs_desc; /* skip internal desc_irqs array and get it from DT */
 };
 
-static struct stm32_exti_host_data *stm32_host_data;
-
 static const struct stm32_exti_bank stm32f4xx_exti_b1 = {
        .imr_ofst       = 0x00,
        .emr_ofst       = 0x04,
        return IRQ_SET_MASK_OK_DONE;
 }
 
-static int __maybe_unused stm32_exti_h_suspend(void)
+static int stm32_exti_h_suspend(struct device *dev)
 {
+       struct stm32_exti_host_data *host_data = dev_get_drvdata(dev);
        struct stm32_exti_chip_data *chip_data;
        int i;
 
-       for (i = 0; i < stm32_host_data->drv_data->bank_nr; i++) {
-               chip_data = &stm32_host_data->chips_data[i];
-               raw_spin_lock(&chip_data->rlock);
+       for (i = 0; i < host_data->drv_data->bank_nr; i++) {
+               chip_data = &host_data->chips_data[i];
                stm32_chip_suspend(chip_data, chip_data->wake_active);
-               raw_spin_unlock(&chip_data->rlock);
        }
 
        return 0;
 }
 
-static void __maybe_unused stm32_exti_h_resume(void)
+static int stm32_exti_h_resume(struct device *dev)
 {
+       struct stm32_exti_host_data *host_data = dev_get_drvdata(dev);
        struct stm32_exti_chip_data *chip_data;
        int i;
 
-       for (i = 0; i < stm32_host_data->drv_data->bank_nr; i++) {
-               chip_data = &stm32_host_data->chips_data[i];
-               raw_spin_lock(&chip_data->rlock);
+       for (i = 0; i < host_data->drv_data->bank_nr; i++) {
+               chip_data = &host_data->chips_data[i];
                stm32_chip_resume(chip_data, chip_data->mask_cache);
-               raw_spin_unlock(&chip_data->rlock);
        }
-}
 
-static struct syscore_ops stm32_exti_h_syscore_ops = {
-#ifdef CONFIG_PM_SLEEP
-       .suspend        = stm32_exti_h_suspend,
-       .resume         = stm32_exti_h_resume,
-#endif
-};
-
-static void stm32_exti_h_syscore_init(struct stm32_exti_host_data *host_data)
-{
-       stm32_host_data = host_data;
-       register_syscore_ops(&stm32_exti_h_syscore_ops);
-}
-
-static void stm32_exti_h_syscore_deinit(void)
-{
-       unregister_syscore_ops(&stm32_exti_h_syscore_ops);
+       return 0;
 }
 
 static int stm32_exti_h_retrigger(struct irq_data *d)
                goto free_chips_data;
        }
 
-       stm32_host_data = host_data;
-
        return host_data;
 
 free_chips_data:
        irq_domain_remove(domain);
 }
 
-static void stm32_exti_remove(struct platform_device *pdev)
-{
-       stm32_exti_h_syscore_deinit();
-}
-
 static int stm32_exti_probe(struct platform_device *pdev)
 {
        int ret, i;
        if (!host_data)
                return -ENOMEM;
 
+       dev_set_drvdata(dev, host_data);
+
        /* check for optional hwspinlock which may be not available yet */
        ret = of_hwspin_lock_get_id(np, 0);
        if (ret == -EPROBE_DEFER)
        if (of_property_read_bool(np, "interrupts-extended"))
                host_data->dt_has_irqs_desc = true;
 
-       stm32_exti_h_syscore_init(host_data);
-
        return 0;
 }
 
 };
 MODULE_DEVICE_TABLE(of, stm32_exti_ids);
 
+static const struct dev_pm_ops stm32_exti_dev_pm_ops = {
+       NOIRQ_SYSTEM_SLEEP_PM_OPS(stm32_exti_h_suspend, stm32_exti_h_resume)
+};
+
 static struct platform_driver stm32_exti_driver = {
        .probe          = stm32_exti_probe,
-       .remove_new     = stm32_exti_remove,
        .driver         = {
                .name           = "stm32_exti",
                .of_match_table = stm32_exti_ids,
+               .pm             = &stm32_exti_dev_pm_ops,
        },
 };