1 files changed, 7 insertions, 5 deletions

diff --git a/drivers/misc/pci_endpoint_test.c b/drivers/misc/pci_endpoint_test.c
index 320276f42653..fe8897e64635 100644
--- a/drivers/misc/pci_endpoint_test.c
+++ b/drivers/misc/pci_endpoint_test.c
@@ -534,12 +534,14 @@ static int pci_endpoint_test_probe(struct pci_dev *pdev,
         }
 
         for (bar = BAR_0; bar <= BAR_5; bar++) {
-                base = pci_ioremap_bar(pdev, bar);
-                if (!base) {
-                        dev_err(dev, "failed to read BAR%d\n", bar);
-                        WARN_ON(bar == test_reg_bar);
+                if (pci_resource_flags(pdev, bar) & IORESOURCE_MEM) {
+                        base = pci_ioremap_bar(pdev, bar);
+                        if (!base) {
+                                dev_err(dev, "failed to read BAR%d\n", bar);
+                                WARN_ON(bar == test_reg_bar);
+                        }
+                        test->bar[bar] = base;
                 }
-                test->bar[bar] = base;
         }
 
         test->base = test->bar[test_reg_bar];