3 files changed, 20 insertions, 14 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
index 5841c72c..87eff813 100644
--- a/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/ce2_gk20a.c
@@ -406,6 +406,7 @@ u32 gk20a_ce_create_context(struct gk20a *g,
 {
         struct gk20a_gpu_ctx *ce_ctx;
         struct gk20a_ce_app *ce_app = &g->ce_app;
+        struct nvgpu_gpfifo_args gpfifo_args;
         u32 ctx_id = ~0;
         int err = 0;
 
@@ -458,8 +459,11 @@ u32 gk20a_ce_create_context(struct gk20a *g,
                 goto end;
         }
 
+        gpfifo_args.num_entries = 1024;
+        gpfifo_args.num_inflight_jobs = 0;
+        gpfifo_args.flags = 0;
         /* allocate gpfifo (1024 should be more than enough) */
-        err = gk20a_channel_alloc_gpfifo(ce_ctx->ch, 1024, 0, 0);
+        err = gk20a_channel_alloc_gpfifo(ce_ctx->ch, &gpfifo_args);
         if (err) {
                 nvgpu_err(g, "ce: unable to allocate gpfifo");
                 goto end;
diff --git a/drivers/gpu/nvgpu/gk20a/channel_gk20a.c b/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
index 8b9f02df..83c49d52 100644
--- a/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/channel_gk20a.c
@@ -1034,9 +1034,7 @@ static void channel_gk20a_free_prealloc_resources(struct channel_gk20a *c)
 }
 
 int gk20a_channel_alloc_gpfifo(struct channel_gk20a *c,
-                unsigned int num_entries,
-                unsigned int num_inflight_jobs,
-                u32 flags)
+                struct nvgpu_gpfifo_args *gpfifo_args)
 {
         struct gk20a *g = c->g;
         struct vm_gk20a *ch_vm;
@@ -1044,13 +1042,13 @@ int gk20a_channel_alloc_gpfifo(struct channel_gk20a *c,
         int err = 0;
         unsigned long acquire_timeout;
 
-        gpfifo_size = num_entries;
+        gpfifo_size = gpfifo_args->num_entries;
         gpfifo_entry_size = nvgpu_get_gpfifo_entry_size();
 
-        if (flags & NVGPU_GPFIFO_FLAGS_SUPPORT_VPR)
+        if (gpfifo_args->flags & NVGPU_GPFIFO_FLAGS_SUPPORT_VPR)
                 c->vpr = true;
 
-        if (flags & NVGPU_GPFIFO_FLAGS_SUPPORT_DETERMINISTIC) {
+        if (gpfifo_args->flags & NVGPU_GPFIFO_FLAGS_SUPPORT_DETERMINISTIC) {
                 nvgpu_rwsem_down_read(&g->deterministic_busy);
                 /*
                  * Railgating isn't deterministic; instead of disallowing
@@ -1137,15 +1135,15 @@ int gk20a_channel_alloc_gpfifo(struct channel_gk20a *c,
 
         err = g->ops.fifo.setup_ramfc(c, c->gpfifo.mem.gpu_va,
                                         c->gpfifo.entry_num,
-                                        acquire_timeout, flags);
+                                        acquire_timeout, gpfifo_args->flags);
         if (err)
                 goto clean_up_sync;
 
         /* TBD: setup engine contexts */
 
-        if (num_inflight_jobs) {
+        if (gpfifo_args->num_inflight_jobs) {
                 err = channel_gk20a_prealloc_resources(c,
-                                num_inflight_jobs);
+                                gpfifo_args->num_inflight_jobs);
                 if (err)
                         goto clean_up_sync;
         }
@@ -1166,7 +1164,7 @@ int gk20a_channel_alloc_gpfifo(struct channel_gk20a *c,
 clean_up_priv_cmd:
         channel_gk20a_free_priv_cmdbuf(c);
 clean_up_prealloc:
-        if (num_inflight_jobs)
+        if (gpfifo_args->num_inflight_jobs)
                 channel_gk20a_free_prealloc_resources(c);
 clean_up_sync:
         if (c->sync) {
diff --git a/drivers/gpu/nvgpu/gk20a/channel_gk20a.h b/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
index f2cb2ab8..eaa5f81b 100644
--- a/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/channel_gk20a.h
@@ -46,6 +46,12 @@ struct fifo_profile_gk20a;
 #define NVGPU_GPFIFO_FLAGS_SUPPORT_DETERMINISTIC        (1 << 1)
 #define NVGPU_GPFIFO_FLAGS_REPLAYABLE_FAULTS_ENABLE     (1 << 2)
 
+struct nvgpu_gpfifo_args {
+        u32 num_entries;
+        u32 num_inflight_jobs;
+        u32 flags;
+};
+
 struct notification {
         struct {
                 u32 nanoseconds[2];
@@ -324,9 +330,7 @@ struct channel_gk20a *gk20a_open_new_channel(struct gk20a *g,
                 bool is_privileged_channel);
 
 int gk20a_channel_alloc_gpfifo(struct channel_gk20a *c,
-                unsigned int num_entries,
-                unsigned int num_inflight_jobs,
-                u32 flags);
+                struct nvgpu_gpfifo_args *gpfifo_args);
 
 void gk20a_channel_timeout_restart_all_channels(struct gk20a *g);