1 files changed, 10 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
index f9884cfb..9cb5b262 100644
--- a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
@@ -211,6 +211,16 @@ int gk20a_tsg_set_timeslice(struct tsg_gk20a *tsg, u32 timeslice)
         return g->ops.fifo.tsg_set_timeslice(tsg, timeslice);
 }
 
+u32 gk20a_tsg_get_timeslice(struct tsg_gk20a *tsg)
+{
+        struct gk20a *g = tsg->g;
+
+        if (!tsg->timeslice_us)
+                return g->ops.fifo.default_timeslice_us(g);
+
+        return tsg->timeslice_us;
+}
+
 static void release_used_tsg(struct fifo_gk20a *f, struct tsg_gk20a *tsg)
 {
         nvgpu_mutex_acquire(&f->tsg_inuse_mutex);