struct pci_dev *pdev  = NULL;
        void __iomem *fb_virt;
        int gen2vm = efi_enabled(EFI_BOOT);
-       resource_size_t pot_start, pot_end;
        phys_addr_t paddr;
        int ret;
 
        dio_fb_size =
                screen_width * screen_height * screen_depth / 8;
 
-       if (gen2vm) {
-               pot_start = 0;
-               pot_end = -1;
-       } else {
-               if (!(pci_resource_flags(pdev, 0) & IORESOURCE_MEM) ||
-                   pci_resource_len(pdev, 0) < screen_fb_size) {
-                       pr_err("Resource not available or (0x%lx < 0x%lx)\n",
-                              (unsigned long) pci_resource_len(pdev, 0),
-                              (unsigned long) screen_fb_size);
-                       goto err1;
-               }
-
-               pot_end = pci_resource_end(pdev, 0);
-               pot_start = pot_end - screen_fb_size + 1;
-       }
-
-       ret = vmbus_allocate_mmio(&par->mem, hdev, pot_start, pot_end,
+       ret = vmbus_allocate_mmio(&par->mem, hdev, 0, -1,
                                  screen_fb_size, 0x100000, true);
        if (ret != 0) {
                pr_err("Unable to allocate framebuffer memory\n");