gpu: nvgpu: Support 512 channels in gm20b

Retrieve channel count from gm20b specific header instead of the
gk20a header. This increases channel count from 128 to 512.

Change-Id: I96d4887432852795f7f526e33f0d3d2458f3af0e
Signed-off-by: Terje Bergstrom <tbergstrom@nvidia.com>
Reviewed-on: http://git-master/r/500623

1 files changed, 7 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
index 3224e29c..0c8bc6f4 100644
--- a/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/fifo_gk20a.c
@@ -513,7 +513,7 @@ static int gk20a_init_fifo_setup_sw(struct gk20a *g)
         mutex_init(&f->intr.isr.mutex);
         gk20a_init_fifo_pbdma_intr_descs(f); /* just filling in data/tables */
 
-        f->num_channels = ccsr_channel__size_1_v();
+        f->num_channels = g->ops.fifo.get_num_fifos(g);
         f->num_pbdma = proj_host_num_pbdma_v();
         f->max_engines = ENGINE_INVAL_GK20A;
 
@@ -2132,6 +2132,11 @@ static void gk20a_fifo_apply_pb_timeout(struct gk20a *g)
         }
 }
 
+static u32 gk20a_fifo_get_num_fifos(struct gk20a *g)
+{
+        return ccsr_channel__size_1_v();
+}
+
 void gk20a_init_fifo(struct gpu_ops *gops)
 {
         gk20a_init_channel(gops);
@@ -2140,4 +2145,5 @@ void gk20a_init_fifo(struct gpu_ops *gops)
         gops->fifo.trigger_mmu_fault = gk20a_fifo_trigger_mmu_fault;
         gops->fifo.apply_pb_timeout = gk20a_fifo_apply_pb_timeout;
         gops->fifo.wait_engine_idle = gk20a_fifo_wait_engine_idle;
+        gops->fifo.get_num_fifos = gk20a_fifo_get_num_fifos;
 }