2 files changed, 21 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c
index b9df58ec..f8f021b7 100644
--- a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c
+++ b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c
@@ -14,13 +14,31 @@
 #include <gk20a/gk20a.h>
 #include <vgpu/gp10b/vgpu_fifo_gp10b.h>
 
+#include "vgpu/vgpu.h"
+
 #include "vgpu_fifo_gv11b.h"
 #include "vgpu_subctx_gv11b.h"
 
+static int vgpu_gv11b_init_fifo_setup_hw(struct gk20a *g)
+{
+        struct fifo_gk20a *f = &g->fifo;
+        int err;
+
+        err = vgpu_get_attribute(vgpu_get_handle(g),
+                        TEGRA_VGPU_ATTRIB_MAX_SUBCTX_COUNT,
+                        &f->t19x.max_subctx_count);
+        if (err) {
+                nvgpu_err(g, "get max_subctx_count failed %d", err);
+                return err;
+        }
+
+        return 0;
+}
+
 void vgpu_gv11b_init_fifo_ops(struct gpu_ops *gops)
 {
         vgpu_gp10b_init_fifo_ops(gops);
 
-        gops->fifo.init_fifo_setup_hw = NULL;
+        gops->fifo.init_fifo_setup_hw = vgpu_gv11b_init_fifo_setup_hw;
         gops->fifo.free_channel_ctx_header = vgpu_gv11b_free_subctx_header;
 }
diff --git a/include/linux/tegra_vgpu_t19x.h b/include/linux/tegra_vgpu_t19x.h
index f473275d..c2814f16 100644
--- a/include/linux/tegra_vgpu_t19x.h
+++ b/include/linux/tegra_vgpu_t19x.h
@@ -31,4 +31,6 @@ union tegra_vgpu_t19x_params {
         struct tegra_vgpu_free_ctx_header_params free_ctx_header;
 };
 
+#define TEGRA_VGPU_ATTRIB_MAX_SUBCTX_COUNT      100
+
 #endif