1 files changed, 19 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;
 }