MpInitLib: Allocate code buffer for PEI phase

[mirror_edk2.git] / UefiCpuPkg / Library / MpInitLib / MpLib.c

diff --git a/UefiCpuPkg/Library/MpInitLib/MpLib.c b/UefiCpuPkg/Library/MpInitLib/MpLib.c
index 4a73787ee43a649548e05964f6b04f24b059c83f..d761bdc4878f5187cb776de53d0883794289662f 100644 (file)
--- a/UefiCpuPkg/Library/MpInitLib/MpLib.c
+++ b/UefiCpuPkg/Library/MpInitLib/MpLib.c
@@ -1056,7 +1056,7 @@ AllocateResetVector (
                                    (CpuMpData->WakeupBuffer +\r
                                     CpuMpData->AddressMap.RendezvousFunnelSize +\r
                                     CpuMpData->AddressMap.SwitchToRealSize);\r
-    CpuMpData->WakeupBufferHigh = GetModeTransitionBuffer (\r
+    CpuMpData->WakeupBufferHigh = AllocateCodeBuffer (\r
                                     CpuMpData->AddressMap.RendezvousFunnelSize +\r
                                     CpuMpData->AddressMap.SwitchToRealSize -\r
                                     CpuMpData->AddressMap.ModeTransitionOffset\r