1 files changed, 55 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
index 849c3f6a..67fa6508 100644
--- a/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/tsg_gk20a.c
@@ -23,6 +23,7 @@
 #include <linux/anon_inodes.h>
 
 #include "gk20a.h"
+#include "hw_ccsr_gk20a.h"
 
 static void gk20a_tsg_release(struct kref *ref);
 
@@ -226,6 +227,60 @@ long gk20a_tsg_dev_ioctl(struct file *filp, unsigned int cmd,
                  */
                 break;
 
+        case NVGPU_IOCTL_TSG_ENABLE:
+                {
+                struct channel_gk20a *ch;
+                err = gk20a_busy(g->dev);
+                if (err) {
+                        gk20a_err(&g->dev->dev,
+                           "failed to host gk20a for ioctl cmd: 0x%x", cmd);
+                        return err;
+                }
+                mutex_lock(&tsg->ch_list_lock);
+                list_for_each_entry(ch, &tsg->ch_list, ch_entry) {
+                        gk20a_writel(ch->g, ccsr_channel_r(ch->hw_chid),
+                                gk20a_readl(ch->g, ccsr_channel_r(ch->hw_chid))
+                                | ccsr_channel_enable_set_true_f());
+                }
+                mutex_unlock(&tsg->ch_list_lock);
+                gk20a_idle(g->dev);
+                break;
+                }
+
+        case NVGPU_IOCTL_TSG_DISABLE:
+                {
+                struct channel_gk20a *ch;
+                err = gk20a_busy(g->dev);
+                if (err) {
+                        gk20a_err(&g->dev->dev,
+                           "failed to host gk20a for ioctl cmd: 0x%x", cmd);
+                        return err;
+                }
+                mutex_lock(&tsg->ch_list_lock);
+                list_for_each_entry(ch, &tsg->ch_list, ch_entry) {
+                        gk20a_writel(ch->g, ccsr_channel_r(ch->hw_chid),
+                                gk20a_readl(ch->g, ccsr_channel_r(ch->hw_chid))
+                                | ccsr_channel_enable_clr_true_f());
+                }
+                mutex_unlock(&tsg->ch_list_lock);
+                gk20a_idle(g->dev);
+                break;
+                }
+
+        case NVGPU_IOCTL_TSG_PREEMPT:
+                {
+                err = gk20a_busy(g->dev);
+                if (err) {
+                        gk20a_err(&g->dev->dev,
+                           "failed to host gk20a for ioctl cmd: 0x%x", cmd);
+                        return err;
+                }
+                /* preempt TSG */
+                err = gk20a_fifo_preempt_tsg(g, tsg->tsgid);
+                gk20a_idle(g->dev);
+                break;
+                }
+
         default:
                 gk20a_err(dev_from_gk20a(g),
                            "unrecognized tsg gpu ioctl cmd: 0x%x",