diff --git a/arch/arm64/mm/mmap.c b/arch/arm64/mm/mmap.c
index ac89686c4af8920c85e1dc3ad1257f386b43a84d..87f29df8126ba231a996de98e3d39d3af5040b24 100644
--- a/arch/arm64/mm/mmap.c
+++ b/arch/arm64/mm/mmap.c
@@ -28,6 +28,7 @@
 #include <linux/io.h>
 #include <linux/personality.h>
 #include <linux/random.h>
+#include <linux/share_pool.h>
 
 #include <asm/cputype.h>
 
@@ -80,7 +81,10 @@ static unsigned long mmap_base(unsigned long rnd, struct rlimit *rlim_stack)
 	else if (gap > MAX_GAP)
 		gap = MAX_GAP;
 
-	return PAGE_ALIGN(STACK_TOP - gap - rnd);
+	if (sp_is_enabled())
+		return ALIGN_DOWN(MMAP_SHARE_POOL_START - rnd, PAGE_SIZE);
+	else
+		return PAGE_ALIGN(STACK_TOP - gap - rnd);
 }
 
 /*
diff --git a/include/linux/share_pool.h b/include/linux/share_pool.h
index 9650f257b3ad7af4f266cf4ca51bc37262564ab8..9557a8be46677d8fe1a5417cc9f1550405ca77b4 100644
--- a/include/linux/share_pool.h
+++ b/include/linux/share_pool.h
@@ -130,8 +130,6 @@ struct sp_proc_stat {
 	atomic64_t k2u_size;
 };
 
-#ifdef CONFIG_ASCEND_SHARE_POOL
-
 #define MAP_SHARE_POOL			0x100000
 
 #define MMAP_TOP_4G_SIZE		0x100000000UL
@@ -148,6 +146,8 @@ struct sp_proc_stat {
 #define MMAP_SHARE_POOL_START		(MMAP_SHARE_POOL_END - MMAP_SHARE_POOL_SIZE)
 #define MMAP_SHARE_POOL_16G_START	(MMAP_SHARE_POOL_END - MMAP_SHARE_POOL_DVPP_SIZE)
 
+#ifdef CONFIG_ASCEND_SHARE_POOL
+
 static inline void sp_init_mm(struct mm_struct *mm)
 {
 	mm->sp_group = NULL;