diff --git a/arch/arm64/mm/init.c b/arch/arm64/mm/init.c
index 88887e0722b7c32dd33813272bbe1fecc318010d..883350f9cc4219bcaec6ff35b957aa9e9b57c747 100644
--- a/arch/arm64/mm/init.c
+++ b/arch/arm64/mm/init.c
@@ -41,6 +41,7 @@
 #include <linux/kexec.h>
 #include <linux/crash_dump.h>
 #include <linux/iommu.h>
+#include <linux/suspend.h>
 
 #include <asm/boot.h>
 #include <asm/fixmap.h>
@@ -779,6 +780,9 @@ static int __init ascend_enable_setup(char *__unused)
 	if (IS_ENABLED(CONFIG_ASCEND_CHARGE_MIGRATE_HUGEPAGES))
 		enable_charge_mighp = 1;
 
+	if (IS_ENABLED(CONFIG_SUSPEND))
+		mem_sleep_current = PM_SUSPEND_ON;
+
 	return 1;
 }