gpu: nvgpu: changes related to preemption

Added function pointers to check chip specific valid
gfx class and compute class. Also added function pointer
to update ctx header with preemption buffer pointers.

Bug 200292090

Change-Id: I8119ee082e2abb67186a8ac07088f8db7f410ba1
Signed-off-by: seshendra Gadagottu <sgadagottu@nvidia.com>
Reviewed-on: http://git-master/r/1293502
Reviewed-by: mobile promotions <svcmobile_promotions@nvidia.com>
Tested-by: mobile promotions <svcmobile_promotions@nvidia.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index c8d06b4f..96ca69a3 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -194,6 +194,8 @@ struct gpu_ops {
                 void (*set_circular_buffer_size)(struct gk20a *g, u32 data);
                 void (*enable_hww_exceptions)(struct gk20a *g);
                 bool (*is_valid_class)(struct gk20a *g, u32 class_num);
+                bool (*is_valid_gfx_class)(struct gk20a *g, u32 class_num);
+                bool (*is_valid_compute_class)(struct gk20a *g, u32 class_num);
                 void (*get_sm_dsm_perf_regs)(struct gk20a *g,
                                                   u32 *num_sm_dsm_perf_regs,
                                                   u32 **sm_dsm_perf_regs,
@@ -354,6 +356,8 @@ struct gpu_ops {
                                         struct nvgpu_mem *mem, u64 gpu_va);
                 void (*write_pm_ptr)(struct gk20a *g,
                                         struct nvgpu_mem *mem, u64 gpu_va);
+                void (*write_preemption_ptr)(struct gk20a *g,
+                                        struct nvgpu_mem *mem, u64 gpu_va);
                 void (*init_elcg_mode)(struct gk20a *g, u32 mode, u32 engine);
                 void (*load_tpc_mask)(struct gk20a *g);
                 int (*inval_icache)(struct gk20a *g, struct channel_gk20a *ch);