1 files changed, 18 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/module.c b/drivers/gpu/nvgpu/os/linux/module.c
index a81d837b..ebe2e650 100644
--- a/drivers/gpu/nvgpu/os/linux/module.c
+++ b/drivers/gpu/nvgpu/os/linux/module.c
@@ -680,6 +680,16 @@ void __iomem *nvgpu_devm_ioremap(struct device *dev, resource_size_t offset,
         return devm_ioremap(dev, offset, size);
 }
 
+u64 nvgpu_resource_addr(struct platform_device *dev, int i)
+{
+        struct resource *r = platform_get_resource(dev, IORESOURCE_MEM, i);
+
+        if (!r)
+                return 0;
+
+        return r->start;
+}
+
 static irqreturn_t gk20a_intr_isr_stall(int irq, void *dev_id)
 {
         struct gk20a *g = dev_id;
@@ -764,6 +774,14 @@ static int gk20a_init_support(struct platform_device *pdev)
                 goto fail;
         }
 
+        l->regs_bus_addr = nvgpu_resource_addr(pdev,
+                        GK20A_BAR0_IORESOURCE_MEM);
+        if (!l->regs_bus_addr) {
+                nvgpu_err(g, "failed to read register bus offset");
+                err = -ENODEV;
+                goto fail;
+        }
+
         l->bar1 = nvgpu_devm_ioremap_resource(pdev,
                                               GK20A_BAR1_IORESOURCE_MEM,
                                               &l->bar1_mem);