UefiCpuPkg/CpuDxe: split out StartupCode from StartApsStackless()

[mirror_edk2.git] / UefiCpuPkg / CpuDxe / CpuMp.c

diff --git a/UefiCpuPkg/CpuDxe/CpuMp.c b/UefiCpuPkg/CpuDxe/CpuMp.c
index 3e30d7417d4f0ea88739911544cfb2984e30ba41..029e8d43317beb026a7726101f219dfb14c80006 100644 (file)
--- a/UefiCpuPkg/CpuDxe/CpuMp.c
+++ b/UefiCpuPkg/CpuDxe/CpuMp.c
@@ -1454,7 +1454,10 @@ InitializeMpSupport (
 \r
   InitMpSystemData ();\r
 \r
+  PrepareAPStartupCode ();\r
+\r
   if (mMpSystemData.NumberOfProcessors == 1) {\r
+    FreeApStartupCode ();\r
     FreePages (mCommonStack, EFI_SIZE_TO_PAGES (gMaxLogicalProcessorNumber * gApStackSize));\r
     return;\r
   }\r