2 files changed, 17 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/hw_ltc_gp10b.h b/drivers/gpu/nvgpu/gp10b/hw_ltc_gp10b.h
index 01b66c9c..ea96a9aa 100644
--- a/drivers/gpu/nvgpu/gp10b/hw_ltc_gp10b.h
+++ b/drivers/gpu/nvgpu/gp10b/hw_ltc_gp10b.h
@@ -514,4 +514,12 @@ static inline u32 ltc_ltc0_lts0_tstg_info_1_slices_per_l2_v(u32 r)
 {
         return (r >> 16) & 0x1f;
 }
+static inline u32 ltc_ltca_g_axi_pctrl_r(void)
+{
+        return 0x00160000;
+}
+static inline u32 ltc_ltca_g_axi_pctrl_user_sid_f(u32 v)
+{
+        return (v & 0xff) << 2;
+}
 #endif
diff --git a/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c b/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
index dbeed474..78c85f3e 100644
--- a/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/ltc_gp10b.c
@@ -148,6 +148,14 @@ void gp10b_ltc_isr(struct gk20a *g)
         }
 }
 
+void gp10b_ltc_init_fs_state(struct gk20a *g)
+{
+        gm20b_ltc_init_fs_state(g);
+
+        gk20a_writel(g, ltc_ltca_g_axi_pctrl_r(),
+                        ltc_ltca_g_axi_pctrl_user_sid_f(31));
+}
+
 void gp10b_init_ltc(struct gpu_ops *gops)
 {
         gops->ltc.determine_L2_size_bytes = gp10b_determine_L2_size_bytes;
@@ -157,7 +165,7 @@ void gp10b_init_ltc(struct gpu_ops *gops)
         gops->ltc.init_cbc = gk20a_ltc_init_cbc;
 
         /* GM20b specific ops. */
-        gops->ltc.init_fs_state = gm20b_ltc_init_fs_state;
+        gops->ltc.init_fs_state = gp10b_ltc_init_fs_state;
         gops->ltc.init_comptags = gp10b_ltc_init_comptags;
         gops->ltc.cbc_ctrl = gm20b_ltc_cbc_ctrl;
         gops->ltc.elpg_flush = gm20b_ltc_g_elpg_flush_locked;