diff --git a/arch/arm64/include/asm/hw_breakpoint.h b/arch/arm64/include/asm/hw_breakpoint.h
index b6b167ac082b69..9b5a831601e0af 100644
--- a/arch/arm64/include/asm/hw_breakpoint.h
+++ b/arch/arm64/include/asm/hw_breakpoint.h
@@ -159,9 +159,15 @@ static inline int get_num_brps(void)
 static inline int get_num_wrps(void)
 {
 	u64 dfr0 = read_system_reg(SYS_ID_AA64DFR0_EL1);
+#ifdef CONFIG_AMLOGIC_VMAP
+	return (1 +
+		cpuid_feature_extract_unsigned_field(dfr0,
+			ID_AA64DFR0_WRPS_SHIFT)) - 2;
+#else
 	return 1 +
 		cpuid_feature_extract_unsigned_field(dfr0,
 						ID_AA64DFR0_WRPS_SHIFT);
+#endif
 }
 
 #endif	/* __KERNEL__ */