diff --git a/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c b/drivers/gpu/nvgpu/vgpu/gv11b/vgpu_fifo_gv11b.c
index b9df58ecb..f8f021b7e 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 f473275de..c2814f16c 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