diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index 4af23e3da..bccd4100f 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -487,6 +487,7 @@ static int gk20a_init_pmu(struct pmu_gk20a *pmu)
 	struct gk20a *g = pmu->g;
 	switch (pmu->desc->app_version) {
 	case APP_VERSION_1:
+	case APP_VERSION_2:
 		g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
 		g->ops.pmu_ver.get_pmu_cmdline_args_size =
 			pmu_cmdline_size_v1;
@@ -618,7 +619,8 @@ static int gk20a_init_pmu(struct pmu_gk20a *pmu)
 		break;
 	default:
 		gk20a_err(dev_from_gk20a(pmu->g),
-		"PMU code version not supported\n");
+		"PMU code version not supported version: %d\n",
+			pmu->desc->app_version);
 		return -EINVAL;
 		break;
 	}
diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
index 73a09db68..8aaea33f7 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
@@ -51,6 +51,7 @@
 /* Mapping between AP_CTRLs and Idle counters */
 #define PMU_AP_IDLE_MASK_GRAPHICS	(PMU_AP_IDLE_MASK_HIST_IDX_1)
 
+#define APP_VERSION_2 18542378
 #define APP_VERSION_1 17997577
 #define APP_VERSION_0 16856675