#include "hyperv_vmbus.h"
 
-static unsigned long virt_to_hvpfn(void *addr)
-{
-       phys_addr_t paddr;
-
-       if (is_vmalloc_addr(addr))
-               paddr = page_to_phys(vmalloc_to_page(addr)) +
-                                        offset_in_page(addr);
-       else
-               paddr = __pa(addr);
-
-       return  paddr >> HV_HYP_PAGE_SHIFT;
-}
-
 /*
  * hv_gpadl_size - Return the real size of a gpadl, the size that Hyper-V uses
  *
 
 
 #include <uapi/linux/hyperv.h>
 
+#include <linux/mm.h>
 #include <linux/types.h>
 #include <linux/scatterlist.h>
 #include <linux/list.h>
 #include <linux/mod_devicetable.h>
 #include <linux/interrupt.h>
 #include <linux/reciprocal_div.h>
+#include <asm/hyperv-tlfs.h>
 
 #define MAX_PAGE_BUFFER_COUNT                          32
 #define MAX_MULTIPAGE_BUFFER_COUNT                     32 /* 128K */
 
 extern struct hyperv_pci_block_ops hvpci_block_ops;
 
+static inline unsigned long virt_to_hvpfn(void *addr)
+{
+       phys_addr_t paddr;
+
+       if (is_vmalloc_addr(addr))
+               paddr = page_to_phys(vmalloc_to_page(addr)) +
+                                    offset_in_page(addr);
+       else
+               paddr = __pa(addr);
+
+       return  paddr >> HV_HYP_PAGE_SHIFT;
+}
+
 #endif /* _HYPERV_H */