2 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/gk20a.h b/drivers/gpu/nvgpu/gk20a/gk20a.h
index c6bc129f..f577f93c 100644
--- a/drivers/gpu/nvgpu/gk20a/gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/gk20a.h
@@ -552,6 +552,7 @@ struct gpu_ops {
                                 struct channel_gk20a *ch);
                 int (*tsg_unbind_channel)(struct channel_gk20a *ch);
                 int (*tsg_open)(struct tsg_gk20a *tsg);
+                void (*tsg_release)(struct tsg_gk20a *tsg);
                 u32 (*eng_runlist_base_size)(void);
                 int (*init_engine_info)(struct fifo_gk20a *f);
                 u32 (*runlist_entry_size)(void);
diff --git a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
index 8906257f..4efcfda8 100644
--- a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
@@ -315,6 +315,9 @@ void gk20a_tsg_release(struct nvgpu_ref *ref)
         struct gk20a *g = tsg->g;
         struct gk20a_event_id_data *event_id_data, *event_id_data_temp;
 
+        if (g->ops.fifo.tsg_release)
+                g->ops.fifo.tsg_release(tsg);
+
         if (tsg->tsg_gr_ctx) {
                 gr_gk20a_free_tsg_gr_ctx(tsg);
                 tsg->tsg_gr_ctx = NULL;