gpu: nvgpu: Allow enabling PC sampling

Allow enabling of PC sampling hardware workaround. It is only
applicable to gm20b.

Bug 1517458
Bug 1573150

Change-Id: Iad6a3ae556489fb7ab9628637d291849d2cd98ea
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/710421

3 files changed, 19 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.c b/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.c
index ffb52549..7cda9949 100644
--- a/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/dbg_gpu_gk20a.c
@@ -377,6 +377,17 @@ static int gk20a_perfbuf_map(struct dbg_session_gk20a *dbg_s,
 static int gk20a_perfbuf_unmap(struct dbg_session_gk20a *dbg_s,
                 struct nvgpu_dbg_gpu_perfbuf_unmap_args *args);
 
+static int gk20a_dbg_pc_sampling(struct dbg_session_gk20a *dbg_s,
+                          struct nvgpu_dbg_gpu_pc_sampling_args *args)
+{
+        struct channel_gk20a *ch = dbg_s->ch;
+        struct gk20a *g = ch->g;
+
+        gk20a_dbg_fn("");
+
+        return g->ops.gr.update_pc_sampling ?
+                g->ops.gr.update_pc_sampling(ch, args->enable) : -EINVAL;
+}
 long gk20a_dbg_gpu_dev_ioctl(struct file *filp, unsigned int cmd,
                              unsigned long arg)
 {
@@ -449,6 +460,11 @@ long gk20a_dbg_gpu_dev_ioctl(struct file *filp, unsigned int cmd,
                        (struct nvgpu_dbg_gpu_perfbuf_unmap_args *)buf);
                 break;
 
+        case NVGPU_DBG_GPU_IOCTL_PC_SAMPLING:
+                err = gk20a_dbg_pc_sampling(dbg_s,
+                           (struct nvgpu_dbg_gpu_pc_sampling_args *)buf);
+                break;
+
         default:
                 gk20a_err(dev_from_gk20a(g),
                            "unrecognized dbg gpu ioctl cmd: 0x%x",
diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index acee2e48..441ea7cc 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -165,6 +165,8 @@ struct gpu_ops {
                                 void *ctx_ptr);
                 int (*dump_gr_regs)(struct gk20a *g,
                                 struct gk20a_debug_output *o);
+                int (*update_pc_sampling)(struct channel_gk20a *ch,
+                                           bool enable);
         } gr;
         const char *name;
         struct {
diff --git a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
index 4b94f863..deafc438 100644
--- a/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gr_gk20a.h
@@ -500,4 +500,5 @@ int gr_gk20a_alloc_gr_ctx(struct gk20a *g,
 void gr_gk20a_free_gr_ctx(struct gk20a *g,
                           struct vm_gk20a *vm, struct gr_ctx_desc *gr_ctx);
 int gr_gk20a_halt_pipe(struct gk20a *g);
+
 #endif /*__GR_GK20A_H__*/