1 files changed, 2 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 276e5c3e..58108d97 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -820,7 +820,8 @@ static int gk20a_pm_finalize_poweron(struct device *dev)
                 goto done;
         }
 
-        err = g->ops.pmu.prepare_ucode(g);
+        if (g->ops.pmu.prepare_ucode)
+                err = g->ops.pmu.prepare_ucode(g);
         if (err) {
                 gk20a_err(dev, "failed to init pmu ucode");
                 goto done;