gpu: nvgpu: add ref counting for GPU sched ctrl

Jira VFND-1968

Change-Id: Id84c5732e312e44db3d412df5c21e429227dd7fa
Signed-off-by: Thomas Fleury <tfleury@nvidia.com>
Reviewed-on: http://git-master/r/1171286
(cherry picked from commit 13a3a4355914635ed175708affef17dc8ef0b133)
Reviewed-on: http://git-master/r/1177824
Reviewed-by: Automatic_Commit_Validation_User
Reviewed-by: Richard Zhao <rizhao@nvidia.com>
GVS: Gerrit_Virtual_Submit
Reviewed-by: Alex Waterman <alexw@nvidia.com>
Reviewed-by: Vijayakumar Subbu <vsubbu@nvidia.com>

1 files changed, 20 insertions, 2 deletions

diff --git a/include/uapi/linux/nvgpu.h b/include/uapi/linux/nvgpu.h
index 17604d32..aa950dfa 100644
--- a/include/uapi/linux/nvgpu.h
+++ b/include/uapi/linux/nvgpu.h
@@ -1576,6 +1576,14 @@ struct nvgpu_sched_tsg_runlist_interleave_args {
         __u32 runlist_interleave;
 };
 
+struct nvgpu_sched_api_version_args {
+        __u32 version;
+};
+
+struct nvgpu_sched_tsg_refcount_args {
+        __u32 tsgid;                    /* in: TSG identifier */
+};
+
 #define NVGPU_SCHED_IOCTL_GET_TSGS                                      \
         _IOWR(NVGPU_SCHED_IOCTL_MAGIC, 1,                               \
                 struct nvgpu_sched_get_tsgs_args)
@@ -1598,9 +1606,17 @@ struct nvgpu_sched_tsg_runlist_interleave_args {
         _IO(NVGPU_SCHED_IOCTL_MAGIC, 7)
 #define NVGPU_SCHED_IOCTL_UNLOCK_CONTROL                                \
         _IO(NVGPU_SCHED_IOCTL_MAGIC, 8)
-
+#define NVGPU_SCHED_IOCTL_GET_API_VERSION                               \
+        _IOR(NVGPU_SCHED_IOCTL_MAGIC, 9,                                \
+            struct nvgpu_sched_api_version_args)
+#define NVGPU_SCHED_IOCTL_GET_TSG                                       \
+        _IOW(NVGPU_SCHED_IOCTL_MAGIC, 10,                               \
+                struct nvgpu_sched_tsg_refcount_args)
+#define NVGPU_SCHED_IOCTL_PUT_TSG                                       \
+        _IOW(NVGPU_SCHED_IOCTL_MAGIC, 11,                               \
+                struct nvgpu_sched_tsg_refcount_args)
 #define NVGPU_SCHED_IOCTL_LAST                                          \
-        _IOC_NR(NVGPU_SCHED_IOCTL_UNLOCK_CONTROL)
+        _IOC_NR(NVGPU_SCHED_IOCTL_PUT_TSG)
 
 #define NVGPU_SCHED_IOCTL_MAX_ARG_SIZE                                  \
         sizeof(struct nvgpu_sched_tsg_get_params_args)
@@ -1620,4 +1636,6 @@ struct nvgpu_sched_event_arg {
         __u64 status;
 };
 
+#define NVGPU_SCHED_API_VERSION         1
+
 #endif