**/\r
\r
#include "PiPei.h"\r
+#include "Platform.h"\r
#include <Library/DebugLib.h>\r
#include <Library/HobLib.h>\r
#include <Library/PeiServicesLib.h>\r
//\r
// Create a memory allocation HOB for the PEI FV.\r
//\r
- // Note: This should be changed to ACPI NVS when S3 resume is enabled.\r
+ // Allocate as ACPI NVS is S3 is supported\r
//\r
BuildMemoryAllocationHob (\r
PcdGet32 (PcdOvmfPeiMemFvBase),\r
PcdGet32 (PcdOvmfPeiMemFvSize),\r
- EfiBootServicesData\r
+ mS3Supported ? EfiACPIMemoryNVS : EfiBootServicesData\r
);\r
\r
//\r