\r
Status = gSmst->SmmAllocatePool (\r
EfiRuntimeServicesData,\r
- gSmst->NumberOfCpus * sizeof (EFI_SMI_CPU_SAVE_STATE),\r
+ gSmst->NumberOfCpus * sizeof (EFI_SMM_CPU_SAVE_STATE),\r
(VOID **)&FrameworkSmst->CpuSaveState\r
);\r
ASSERT_EFI_ERROR (Status);\r
- ZeroMem (FrameworkSmst->CpuSaveState, gSmst->NumberOfCpus * sizeof (EFI_SMI_CPU_SAVE_STATE));\r
+ ZeroMem (FrameworkSmst->CpuSaveState, gSmst->NumberOfCpus * sizeof (EFI_SMM_CPU_SAVE_STATE));\r
\r
///\r
/// Do not support floating point state now\r
///\r
Status = mSmmCpu->ReadSaveState (\r
mSmmCpu,\r
- EFI_SMM_SAVE_STATE_IO_WIDTH_UINT32,\r
+ (UINTN)sizeof (UINT32),\r
mCpuSaveStateConvTable[Index].Register,\r
CpuIndex,\r
((UINT8 *)SaveState) + mCpuSaveStateConvTable[Index].Offset\r
for (Index = 0; Index < sizeof (mCpuSaveStateConvTable) / sizeof (CPU_SAVE_STATE_CONVERSION); Index++) {\r
Status = mSmmCpu->WriteSaveState (\r
mSmmCpu,\r
- EFI_SMM_SAVE_STATE_IO_WIDTH_UINT32,\r
+ (UINTN)sizeof (UINT32),\r
mCpuSaveStateConvTable[Index].Register,\r
CpuIndex,\r
((UINT8 *)&mFrameworkSmst->CpuSaveState[CpuIndex].Ia32SaveState) + \r