1 files changed, 4 insertions, 1 deletions

diff --git a/arch/x86/pci/amd_bus.c b/arch/x86/pci/amd_bus.c
index a48be98e9ded..f05cbf0764d3 100644
--- a/arch/x86/pci/amd_bus.c
+++ b/arch/x86/pci/amd_bus.c
@@ -380,10 +380,13 @@ static int __init pci_io_ecs_init(void)
         if (early_pci_allowed())
                 pci_enable_pci_io_ecs();
 
-        register_cpu_notifier(&amd_cpu_notifier);
+        cpu_notifier_register_begin();
         for_each_online_cpu(cpu)
                 amd_cpu_notify(&amd_cpu_notifier, (unsigned long)CPU_ONLINE,
                                (void *)(long)cpu);
+        __register_cpu_notifier(&amd_cpu_notifier);
+        cpu_notifier_register_done();
+
         pci_probe |= PCI_HAS_IO_ECS;
 
         return 0;