2 files changed, 4 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index a1e0be19..b0940229 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -1442,6 +1442,7 @@ int gk20a_init_pmu(struct pmu_gk20a *pmu)
                 break;
         case APP_VERSION_GM206:
         case APP_VERSION_NV_GPU:
+        case APP_VERSION_NV_GPU_1:
                 g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
                                 pg_cmd_eng_buf_load_size_v2;
                 g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
@@ -1500,7 +1501,8 @@ int gk20a_init_pmu(struct pmu_gk20a *pmu)
                 g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
                         pmu_allocation_set_dmem_offset_v3;
 
-                if(pmu->desc->app_version != APP_VERSION_NV_GPU) {
+                if(pmu->desc->app_version != APP_VERSION_NV_GPU &&
+                        pmu->desc->app_version != APP_VERSION_NV_GPU_1) {
                         g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
                                         get_pmu_init_msg_pmu_queue_params_v2;
                         g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
index 85f1ffc1..bbcccdeb 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
@@ -54,6 +54,7 @@
 #define APP_VERSION_NC_0        20360931
 #define APP_VERSION_GM206       20652057
 #define APP_VERSION_NV_GPU      20660622
+#define APP_VERSION_NV_GPU_1    20799797
 #define APP_VERSION_GM20B_5 20490253
 #define APP_VERSION_GM20B_4 19008461
 #define APP_VERSION_GM20B_3 18935575