1 files changed, 20 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/gp10b_gating_reglist.c b/drivers/gpu/nvgpu/gp10b/gp10b_gating_reglist.c
index 6bcd9f35..4719b13e 100644
--- a/drivers/gpu/nvgpu/gp10b/gp10b_gating_reglist.c
+++ b/drivers/gpu/nvgpu/gp10b/gp10b_gating_reglist.c
@@ -163,6 +163,11 @@ static const struct gating_desc gp10b_blcg_bus[] = {
         {.addr = 0x00001c00, .prod = 0x00000042, .disable = 0x00000000},
 };
 
+/* blcg ce */
+static const struct gating_desc gp10b_blcg_ce[] = {
+        {.addr = 0x00104200, .prod = 0x00008242, .disable = 0x00000000},
+};
+
 /* blcg ctxsw prog */
 static const struct gating_desc gp10b_blcg_ctxsw_prog[] = {
 };
@@ -483,6 +488,21 @@ void gp10b_blcg_bus_load_gating_prod(struct gk20a *g,
         }
 }
 
+void gp10b_blcg_ce_load_gating_prod(struct gk20a *g,
+        bool prod)
+{
+        u32 i;
+        u32 size = sizeof(gp10b_blcg_ce) / sizeof(struct gating_desc);
+        for (i = 0; i < size; i++) {
+                if (prod)
+                        gk20a_writel(g, gp10b_blcg_ce[i].addr,
+                                gp10b_blcg_ce[i].prod);
+                else
+                        gk20a_writel(g, gp10b_blcg_ce[i].addr,
+                                gp10b_blcg_ce[i].disable);
+        }
+}
+
 void gp10b_blcg_ctxsw_firmware_load_gating_prod(struct gk20a *g,
         bool prod)
 {