4 files changed, 30 insertions, 3 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
index 6b593359..5630e406 100644
--- a/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
+++ b/drivers/gpu/nvgpu/vgpu/gp10b/vgpu_hal_gp10b.c
@@ -139,7 +139,7 @@ static const struct gpu_ops vgpu_gp10b_ops = {
                 .update_ctxsw_preemption_mode =
                         gr_gp10b_update_ctxsw_preemption_mode,
                 .dump_gr_regs = NULL,
-                .update_pc_sampling = gr_gm20b_update_pc_sampling,
+                .update_pc_sampling = vgpu_gr_update_pc_sampling,
                 .get_fbp_en_mask = vgpu_gr_get_fbp_en_mask,
                 .get_max_ltc_per_fbp = vgpu_gr_get_max_ltc_per_fbp,
                 .get_max_lts_per_ltc = vgpu_gr_get_max_lts_per_ltc,
diff --git a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
index 1e633d5f..9d765984 100644
--- a/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/gr_vgpu.c
@@ -1310,3 +1310,30 @@ int vgpu_gr_init_fs_state(struct gk20a *g)
 
         return g->ops.gr.init_sm_id_table(g);
 }
+
+int vgpu_gr_update_pc_sampling(struct channel_gk20a *ch, bool enable)
+{
+        struct tegra_vgpu_cmd_msg msg;
+        struct tegra_vgpu_channel_update_pc_sampling *p =
+                                                &msg.params.update_pc_sampling;
+        struct gk20a *g;
+        int err = -EINVAL;
+
+        if (!ch->g)
+                return err;
+        g = ch->g;
+        nvgpu_log_fn(g, " ");
+
+        msg.cmd = TEGRA_VGPU_CMD_UPDATE_PC_SAMPLING;
+        msg.handle = vgpu_get_handle(g);
+        p->handle = ch->virt_ctx;
+        if (enable)
+                p->mode = TEGRA_VGPU_ENABLE_SAMPLING;
+        else
+                p->mode = TEGRA_VGPU_DISABLE_SAMPLING;
+
+        err = vgpu_comm_sendrecv(&msg, sizeof(msg), sizeof(msg));
+        WARN_ON(err || msg.ret);
+
+        return err ? err : msg.ret;
+}
diff --git a/drivers/gpu/nvgpu/vgpu/gr_vgpu.h b/drivers/gpu/nvgpu/vgpu/gr_vgpu.h
index d6b25b97..c4b3944e 100644
--- a/drivers/gpu/nvgpu/vgpu/gr_vgpu.h
+++ b/drivers/gpu/nvgpu/vgpu/gr_vgpu.h
@@ -70,5 +70,5 @@ int vgpu_gr_resume_contexts(struct gk20a *g,
 int vgpu_gr_commit_inst(struct channel_gk20a *c, u64 gpu_va);
 int vgpu_gr_init_sm_id_table(struct gk20a *g);
 int vgpu_gr_init_fs_state(struct gk20a *g);
-
+int vgpu_gr_update_pc_sampling(struct channel_gk20a *ch, bool enable);
 #endif
diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
index 1b78a4c9..7b536329 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_hal_gv11b.c
@@ -156,7 +156,7 @@ static const struct gpu_ops vgpu_gv11b_ops = {
                 .update_ctxsw_preemption_mode =
                         gr_gv11b_update_ctxsw_preemption_mode,
                 .dump_gr_regs = NULL,
-                .update_pc_sampling = gr_gm20b_update_pc_sampling,
+                .update_pc_sampling = vgpu_gr_update_pc_sampling,
                 .get_fbp_en_mask = vgpu_gr_get_fbp_en_mask,
                 .get_max_ltc_per_fbp = vgpu_gr_get_max_ltc_per_fbp,
                 .get_max_lts_per_ltc = vgpu_gr_get_max_lts_per_ltc,