1 files changed, 5 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gp106/bios_gp106.c b/drivers/gpu/nvgpu/gp106/bios_gp106.c
index e30baf38..09d7b9af 100644
--- a/drivers/gpu/nvgpu/gp106/bios_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/bios_gp106.c
@@ -196,7 +196,9 @@ int gp106_bios_init(struct gk20a *g)
         g->bios.data = nvgpu_vmalloc(g, BIOS_SIZE);
         if (!g->bios.data)
                 return -ENOMEM;
-        g->ops.xve.disable_shadow_rom(g);
+
+        if (g->ops.xve.disable_shadow_rom)
+                g->ops.xve.disable_shadow_rom(g);
         for (i = 0; i < g->bios.size/4; i++) {
                 u32 val = be32_to_cpu(gk20a_readl(g, 0x300000 + i*4));
 
@@ -205,7 +207,8 @@ int gp106_bios_init(struct gk20a *g)
                 g->bios.data[(i*4)+2] = (val >> 8) & 0xff;
                 g->bios.data[(i*4)+3] = val & 0xff;
         }
-        g->ops.xve.enable_shadow_rom(g);
+        if (g->ops.xve.enable_shadow_rom)
+                g->ops.xve.enable_shadow_rom(g);
 
         err = nvgpu_bios_parse_rom(g);
         if (err)