diff --git a/arch/arm/include/asm/memory.h b/arch/arm/include/asm/memory.h
index 5ecadde3b02c..bde40c41c574 100644
--- a/arch/arm/include/asm/memory.h
+++ b/arch/arm/include/asm/memory.h
@@ -263,9 +263,16 @@ static inline unsigned long __phys_to_virt(phys_addr_t x)
 
 #endif
 
+#ifdef CONFIG_AMLOGIC_MODIFY
+#define virt_to_pfn(kaddr) \
+	({virt_check((unsigned long)kaddr); \
+	 ((((unsigned long)(kaddr) - PAGE_OFFSET) >> PAGE_SHIFT) + \
+	 PHYS_PFN_OFFSET); })
+#else
 #define virt_to_pfn(kaddr) \
 	((((unsigned long)(kaddr) - PAGE_OFFSET) >> PAGE_SHIFT) + \
 	 PHYS_PFN_OFFSET)
+#endif
 
 /*
  * These are *only* valid on the kernel direct mapped RAM memory.