aboutsummaryrefslogtreecommitdiff
path: root/src/lib/imd_cbmem.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/imd_cbmem.c')
-rw-r--r--src/lib/imd_cbmem.c5
1 files changed, 2 insertions, 3 deletions
diff --git a/src/lib/imd_cbmem.c b/src/lib/imd_cbmem.c
index e5c3e417d3..f04e0862fe 100644
--- a/src/lib/imd_cbmem.c
+++ b/src/lib/imd_cbmem.c
@@ -172,14 +172,13 @@ int cbmem_initialize_id_size(u32 id, u64 size)
if (imd_recover(imd))
return 1;
-#if defined(__PRE_RAM__)
/*
* Lock the imd in romstage on a recovery. The assumption is that
* if the imd area was recovered in romstage then S3 resume path
* is being taken.
*/
- imd_lockdown(imd);
-#endif
+ if (ENV_ROMSTAGE)
+ imd_lockdown(imd);
/* Add the specified range first */
if (size)