1 files changed, 30 insertions, 15 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
index 3f03e25a..252b4e43 100644
--- a/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/fifo_vgpu.c
@@ -163,27 +163,42 @@ static int vgpu_channel_setup_ramfc(struct channel_gk20a *ch, u64 gpfifo_base,
 
 static int vgpu_fifo_init_engine_info(struct fifo_gk20a *f)
 {
-        struct fifo_engine_info_gk20a *gr_info;
-        struct fifo_engine_info_gk20a *ce_info;
-        const u32 gr_sw_id = ENGINE_GR_GK20A;
-        const u32 ce_sw_id = ENGINE_GRCE_GK20A;
+        struct vgpu_priv_data *priv = vgpu_get_priv_data(f->g);
+        struct tegra_vgpu_engines_info *engines = &priv->constants.engines_info;
+        u32 i;
 
         gk20a_dbg_fn("");
 
-        f->num_engines = 2;
+        if (engines->num_engines > TEGRA_VGPU_MAX_ENGINES) {
+                nvgpu_err(f->g, "num_engines %d larger than max %d",
+                        engines->num_engines, TEGRA_VGPU_MAX_ENGINES);
+                return -EINVAL;
+        }
+
+        f->num_engines = engines->num_engines;
+        for (i = 0; i < f->num_engines; i++) {
+                struct fifo_engine_info_gk20a *info =
+                                        &f->engine_info[engines->info[i].engine_id];
 
-        gr_info = &f->engine_info[0];
+                if (engines->info[i].engine_id >= f->max_engines) {
+                        nvgpu_err(f->g, "engine id %d larger than max %d",
+                                engines->info[i].engine_id,
+                                f->max_engines);
+                        return -EINVAL;
+                }
 
-        /* FIXME: retrieve this from server */
-        gr_info->runlist_id = 0;
-        gr_info->engine_enum = gr_sw_id;
-        f->active_engines_list[0] = 0;
+                info->intr_mask = engines->info[i].intr_mask;
+                info->reset_mask = engines->info[i].reset_mask;
+                info->runlist_id = engines->info[i].runlist_id;
+                info->pbdma_id = engines->info[i].pbdma_id;
+                info->inst_id = engines->info[i].inst_id;
+                info->pri_base = engines->info[i].pri_base;
+                info->engine_enum = engines->info[i].engine_enum;
+                info->fault_id = engines->info[i].fault_id;
+                f->active_engines_list[i] = engines->info[i].engine_id;
+        }
 
-        ce_info = &f->engine_info[1];
-        ce_info->runlist_id = 0;
-        ce_info->inst_id = 2;
-        ce_info->engine_enum = ce_sw_id;
-        f->active_engines_list[1] = 1;
+        gk20a_dbg_fn("done");
 
         return 0;
 }