gpu: nvgpu: Add ioctl to create new TSG

Add ioctl to nvhost-ctrl to create a new TSG.

Bug 200042993

Change-Id: Icdd0edb1d9e374740ace6da9eb3a10c57c62617a
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>

1 files changed, 44 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
index fbc93fc3..93831844 100644
--- a/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ctrl_gk20a.c
@@ -18,6 +18,7 @@
 #include <linux/cdev.h>
 #include <linux/file.h>
 #include <linux/anon_inodes.h>
+#include <linux/nvgpu.h>
 #include <uapi/linux/nvgpu.h>
 
 #include "gk20a.h"
@@ -181,6 +182,45 @@ clean_up:
         return err;
 }
 
+static int gk20a_ctrl_open_tsg(struct gk20a *g,
+                               struct nvgpu_gpu_open_tsg_args *args)
+{
+        struct platform_device *dev = g->dev;
+        int err;
+        int fd;
+        struct file *file;
+        char *name;
+
+        err = get_unused_fd_flags(O_RDWR);
+        if (err < 0)
+                return err;
+        fd = err;
+
+        name = kasprintf(GFP_KERNEL, "nvgpu-%s-tsg%d",
+                         dev_name(&dev->dev), fd);
+
+        file = anon_inode_getfile(name, g->tsg.cdev.ops, NULL, O_RDWR);
+        kfree(name);
+        if (IS_ERR(file)) {
+                err = PTR_ERR(file);
+                goto clean_up;
+        }
+        fd_install(fd, file);
+
+        err = gk20a_tsg_open(g, file);
+        if (err)
+                goto clean_up_file;
+
+        args->tsg_fd = fd;
+        return 0;
+
+clean_up_file:
+        fput(file);
+clean_up:
+        put_unused_fd(fd);
+        return err;
+}
+
 long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
 {
         struct platform_device *dev = filp->private_data;
@@ -346,6 +386,10 @@ long gk20a_ctrl_dev_ioctl(struct file *filp, unsigned int cmd, unsigned long arg
                 err = gk20a_ctrl_alloc_as(g,
                         (struct nvgpu_alloc_as_args *)buf);
                 break;
+        case NVGPU_GPU_IOCTL_OPEN_TSG:
+                err = gk20a_ctrl_open_tsg(g,
+                        (struct nvgpu_gpu_open_tsg_args *)buf);
+                break;
         default:
                 dev_dbg(dev_from_gk20a(g), "unrecognized gpu ioctl cmd: 0x%x", cmd);
                 err = -ENOTTY;