summaryrefslogtreecommitdiff
path: root/UnixPkg/CpuRuntimeDxe
diff options
context:
space:
mode:
Diffstat (limited to 'UnixPkg/CpuRuntimeDxe')
-rw-r--r--UnixPkg/CpuRuntimeDxe/Cpu.c3
-rw-r--r--UnixPkg/CpuRuntimeDxe/CpuIo.c2
2 files changed, 2 insertions, 3 deletions
diff --git a/UnixPkg/CpuRuntimeDxe/Cpu.c b/UnixPkg/CpuRuntimeDxe/Cpu.c
index 689a50a704..dc25d4d812 100644
--- a/UnixPkg/CpuRuntimeDxe/Cpu.c
+++ b/UnixPkg/CpuRuntimeDxe/Cpu.c
@@ -21,7 +21,7 @@ Abstract:
in this driver.
--*/
-#include "PiDxe.h"
+#include <FrameworkDxe.h>
#include <Protocol/Cpu.h>
#include <Protocol/DataHub.h>
#include <Guid/DataHubRecords.h>
@@ -36,7 +36,6 @@ Abstract:
#include <Library/BaseMemoryLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/UefiBootServicesTableLib.h>
-#include <Framework/DataHubRecords.h>
#include "CpuDriver.h"
#include "UnixDxe.h"
#include <Protocol/UnixIo.h>
diff --git a/UnixPkg/CpuRuntimeDxe/CpuIo.c b/UnixPkg/CpuRuntimeDxe/CpuIo.c
index 5c3ebb2f38..5980a03d44 100644
--- a/UnixPkg/CpuRuntimeDxe/CpuIo.c
+++ b/UnixPkg/CpuRuntimeDxe/CpuIo.c
@@ -22,7 +22,7 @@ Abstract:
Bridge I/O Protocol.
--*/
-#include "PiDxe.h"
+#include <FrameworkDxe.h>
#include <Protocol/Cpu.h>
#include <Protocol/DataHub.h>
#include <Guid/DataHubRecords.h>