2 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 2ae1b758..f1dc2f03 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -436,6 +436,7 @@ struct gpu_ops {
                 void (*init_gfxp_wfi_timeout_count)(struct gk20a *g);
                 unsigned long (*get_max_gfxp_wfi_timeout_count)
                                         (struct gk20a *g);
+                void (*ecc_init_scrub_reg)(struct gk20a *g);
         } gr;
         struct {
                 void (*init_hw)(struct gk20a *g);
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
index 99f85b57..ea4d1d24 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.c
@@ -4478,7 +4478,10 @@ static int gk20a_init_gr_setup_hw(struct gk20a *g)
         if (g->ops.gr.enable_gpc_exceptions)
                 g->ops.gr.enable_gpc_exceptions(g);
 
-        /* TBD: ECC for L1/SM */
+        /* enable ECC for L1/SM */
+        if (g->ops.gr.ecc_init_scrub_reg)
+                g->ops.gr.ecc_init_scrub_reg(g);
+
         /* TBD: enable per BE exceptions */
 
         /* reset and enable exceptions */