gpu: nvgpu: vgpu: add tsg set timeslice support

Bug 1702773
JIRA VFND-1496

Change-Id: Ice570df78d974fa59f2a932caf0e6249b13493a1
Signed-off-by: Richard Zhao <rizhao@nvidia.com>
Reviewed-on: http://git-master/r/1144929
(cherry picked from commit 8b6ec996f3773e497a040a8fe4148e01e8dc35fa)
Reviewed-on: http://git-master/r/1150705
Reviewed-by: Automatic_Commit_Validation_User
GVS: Gerrit_Virtual_Submit
Reviewed-by: Terje Bergstrom <tbergstrom@nvidia.com>

2 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index 6f47f228..36bc0fb7 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -340,6 +340,7 @@ struct gpu_ops {
                                         u32 new_level);
                 int (*channel_set_timeslice)(struct channel_gk20a *ch,
                                         u32 timeslice);
+                int (*tsg_set_timeslice)(struct tsg_gk20a *tsg, u32 timeslice);
                 int (*force_reset_ch)(struct channel_gk20a *ch, bool verbose);
                 int (*engine_enum_from_type)(struct gk20a *g, u32 engine_type,
                                         u32 *inst_id);
diff --git a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
index 1e479395..5298d380 100644
--- a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
@@ -582,7 +582,7 @@ long gk20a_tsg_dev_ioctl(struct file *filp, unsigned int cmd,
                            "failed to host gk20a for ioctl cmd: 0x%x", cmd);
                         return err;
                 }
-                err = gk20a_tsg_set_timeslice(tsg,
+                err = g->ops.fifo.tsg_set_timeslice(tsg,
                         ((struct nvgpu_timeslice_args *)buf)->timeslice_us);
                 gk20a_idle(g->dev);
                 break;
@@ -607,4 +607,5 @@ void gk20a_init_tsg_ops(struct gpu_ops *gops)
 {
         gops->fifo.tsg_bind_channel = gk20a_tsg_bind_channel;
         gops->fifo.tsg_unbind_channel = gk20a_tsg_unbind_channel;
+        gops->fifo.tsg_set_timeslice = gk20a_tsg_set_timeslice;
 }