4 files changed, 19 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/vgpu/ce2_vgpu.c b/drivers/gpu/nvgpu/vgpu/ce2_vgpu.c
index f7b06f3b..81035d71 100644
--- a/drivers/gpu/nvgpu/vgpu/ce2_vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/ce2_vgpu.c
@@ -1,7 +1,7 @@
 /*
  * Virtualized GPU CE2
  *
- * Copyright (c) 2015, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2015-2017, NVIDIA CORPORATION.  All rights reserved.
  *
  * This program is free software; you can redistribute it and/or modify it
  * under the terms and conditions of the GNU General Public License,
@@ -33,3 +33,15 @@ int vgpu_ce2_nonstall_isr(struct gk20a *g,
 
         return 0;
 }
+
+static u32 vgpu_ce_get_num_pce(struct gk20a *g)
+{
+        struct vgpu_priv_data *priv = vgpu_get_priv_data(g);
+
+        return priv->constants.num_pce;
+}
+
+void vgpu_init_ce2_ops(struct gpu_ops *gops)
+{
+        gops->ce2.get_num_pce = vgpu_ce_get_num_pce;
+}
diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.c b/drivers/gpu/nvgpu/vgpu/vgpu.c
index 8f401ec6..06971bb0 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.c
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.c
@@ -440,6 +440,7 @@ void vgpu_init_hal_common(struct gk20a *g)
 #if defined(CONFIG_GK20A_CYCLE_STATS)
         vgpu_init_css_ops(gops);
 #endif
+        vgpu_init_ce2_ops(gops);
         gops->chip_init_gpu_characteristics = vgpu_init_gpu_characteristics;
         gops->bus.read_ptimer = vgpu_read_ptimer;
         gops->bus.get_timestamps_zipper = vgpu_get_timestamps_zipper;
diff --git a/drivers/gpu/nvgpu/vgpu/vgpu.h b/drivers/gpu/nvgpu/vgpu/vgpu.h
index cee81bb1..c5e57efa 100644
--- a/drivers/gpu/nvgpu/vgpu/vgpu.h
+++ b/drivers/gpu/nvgpu/vgpu/vgpu.h
@@ -94,6 +94,7 @@ void vgpu_init_tsg_ops(struct gpu_ops *gops);
 #if defined(CONFIG_GK20A_CYCLE_STATS)
 void vgpu_init_css_ops(struct gpu_ops *gops);
 #endif
+void vgpu_init_ce2_ops(struct gpu_ops *gops);
 int vgpu_init_mm_support(struct gk20a *g);
 int vgpu_init_gr_support(struct gk20a *g);
 int vgpu_init_fifo_support(struct gk20a *g);
@@ -178,6 +179,9 @@ static inline void vgpu_init_css_ops(struct gpu_ops *gops)
 {
 }
 #endif
+static inline void vgpu_init_ce2_ops(struct gpu_ops *gops)
+{
+}
 static inline int vgpu_init_mm_support(struct gk20a *g)
 {
         return -ENOSYS;
diff --git a/include/linux/tegra_vgpu.h b/include/linux/tegra_vgpu.h
index 1c486f5c..bfa5862b 100644
--- a/include/linux/tegra_vgpu.h
+++ b/include/linux/tegra_vgpu.h
@@ -477,6 +477,7 @@ struct tegra_vgpu_constants_params {
         u32 preempt_ctx_size;
         u32 channel_base;
         struct tegra_vgpu_engines_info engines_info;
+        u32 num_pce;
 };
 
 struct tegra_vgpu_channel_cyclestats_snapshot_params {