2 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.c b/drivers/gpu/nvgpu/gk20a/gk20a.c
index 7d577cda..7bd6acfd 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.c
@@ -332,7 +332,7 @@ int gk20a_finalize_poweron(struct gk20a *g)
                 if (platform->disable_aspm && g->ops.xve.disable_aspm)
                         g->ops.xve.disable_aspm(g);
 
-                g->ops.xve.sw_init(dev_from_gk20a(g));
+                g->ops.xve.sw_init(g);
                 g->ops.xve.available_speeds(g, &speed);
 
                 /* Set to max speed */
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 83cdeb3f..aeba1fcd 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -969,7 +969,7 @@ struct gpu_ops {
         } css;
 #endif
         struct {
-                int (*sw_init)(struct device *dev);
+                int (*sw_init)(struct gk20a *g);
                 int (*get_speed)(struct gk20a *g, u32 *xve_link_speed);
                 int (*set_speed)(struct gk20a *g, u32 xve_link_speed);
                 void (*available_speeds)(struct gk20a *g, u32 *speed_mask);