3 files changed, 29 insertions, 5 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/ioctl_tsg_t19x.c b/drivers/gpu/nvgpu/common/linux/ioctl_tsg_t19x.c
index 797115ac..ec7501ce 100644
--- a/drivers/gpu/nvgpu/common/linux/ioctl_tsg_t19x.c
+++ b/drivers/gpu/nvgpu/common/linux/ioctl_tsg_t19x.c
@@ -28,6 +28,7 @@ static int gv11b_tsg_ioctl_bind_channel_ex(struct gk20a *g,
         struct nvgpu_os_linux *l = nvgpu_os_linux_from_gk20a(g);
         struct gk20a_sched_ctrl *sched = &l->sched_ctrl;
         struct channel_gk20a *ch;
+        struct gr_gk20a *gr = &g->gr;
         int err = 0;
 
         nvgpu_log(g, gpu_dbg_fn | gpu_dbg_sched, "tsgid=%u", tsg->tsgid);
@@ -48,6 +49,22 @@ static int gv11b_tsg_ioctl_bind_channel_ex(struct gk20a *g,
                 err = -EINVAL;
                 goto idle;
         }
+
+        if (arg->tpc_pg_enabled && (!tsg->t19x.tpc_num_initialized)) {
+                if ((arg->num_active_tpcs > gr->max_tpc_count) ||
+                                !(arg->num_active_tpcs)) {
+                        nvgpu_err(g, "Invalid num of active TPCs");
+                        err = -EINVAL;
+                        goto ch_put;
+                }
+                tsg->t19x.tpc_num_initialized = true;
+                tsg->t19x.num_active_tpcs = arg->num_active_tpcs;
+                tsg->t19x.tpc_pg_enabled = true;
+        } else {
+                tsg->t19x.tpc_pg_enabled = false;
+                nvgpu_log(g, gpu_dbg_info, "dynamic TPC-PG not enabled");
+        }
+
         if (arg->subcontext_id < g->fifo.t19x.max_subctx_count) {
                 ch->t19x.subctx_id = arg->subcontext_id;
         } else {
diff --git a/drivers/gpu/nvgpu/tsg_t19x.h b/drivers/gpu/nvgpu/tsg_t19x.h
index 18c00821..d1f47cc3 100644
--- a/drivers/gpu/nvgpu/tsg_t19x.h
+++ b/drivers/gpu/nvgpu/tsg_t19x.h
@@ -22,9 +22,15 @@
  * DEALINGS IN THE SOFTWARE.
  */
 
-#ifndef _NVGPU_TSG_T19X_H_
-#define _NVGPU_TSG_T19X_H_
+#ifndef __NVGPU_TSG_T19X_H__
+#define __NVGPU_TSG_T19X_H__
 
-#include "common/linux/ioctl_tsg_t19x.h"
+#include <nvgpu/types.h>
+
+struct tsg_t19x {
+        u32  num_active_tpcs;
+        u8   tpc_pg_enabled;
+        bool tpc_num_initialized;
+};
 
 #endif
diff --git a/include/uapi/linux/nvgpu-t19x.h b/include/uapi/linux/nvgpu-t19x.h
index f56bc9d7..27db97c0 100644
--- a/include/uapi/linux/nvgpu-t19x.h
+++ b/include/uapi/linux/nvgpu-t19x.h
@@ -44,8 +44,9 @@ struct nvgpu_tsg_bind_channel_ex_args {
 
         /* in: VEID in Volta */
         __u32 subcontext_id;
-
-        __u64 reserved[2];
+        __u32 num_active_tpcs;
+        __u8 tpc_pg_enabled;
+        __u8 reserved[11];
 };
 
 #define NVGPU_TSG_IOCTL_BIND_CHANNEL_EX \