gpu: nvgpu: reorganize PMU F/W support

- Moved pmu f/w related support from pmu_gk20a.c
to "drivers/gpu/nvgpu/common/pmu/pmu_fw.c" file
- Prepended with nvgpu_ for global functions & replaced
wherever used
- Moved below list related to PMU f/w
  init/remove,
  PMU version specific ops,
  non-secure ucode blob prepare,

JIRA NVGPU-56

Change-Id: Ifdad8c560bd233e98728717d5868119e9d8e8d90
Signed-off-by: Mahantesh Kumbar <mkumbar@nvidia.com>
Reviewed-on: http://git-master/r/1480636
GVS: Gerrit_Virtual_Submit
Reviewed-by: Vijayakumar Subbu <vsubbu@nvidia.com>
Reviewed-by: svccoveritychecker <svccoveritychecker@nvidia.com>

7 files changed, 2305 insertions, 2189 deletions

diff --git a/drivers/gpu/nvgpu/Makefile.nvgpu b/drivers/gpu/nvgpu/Makefile.nvgpu
index f398e384..77d7be4a 100644
--- a/drivers/gpu/nvgpu/Makefile.nvgpu
+++ b/drivers/gpu/nvgpu/Makefile.nvgpu
@@ -60,6 +60,7 @@ nvgpu-y := \
         common/falcon/falcon.o \
         common/pmu/pmu.o \
         common/pmu/pmu_ipc.o \
+        common/pmu/pmu_fw.o \
         gk20a/gk20a.o \
         gk20a/bus_gk20a.o \
         gk20a/pramin_gk20a.o \
diff --git a/drivers/gpu/nvgpu/common/pmu/pmu_fw.c b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
new file mode 100644
index 00000000..f6229a3a
--- /dev/null
+++ b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
@@ -0,0 +1,2291 @@
+/*
+ * Copyright (c) 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,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ */
+
+#include <nvgpu/pmu.h>
+#include <nvgpu/dma.h>
+#include <nvgpu/log.h>
+#include <nvgpu/pmuif/nvgpu_gpmu_cmdif.h>
+#include <nvgpu/firmware.h>
+
+#include "gk20a/gk20a.h"
+
+/* PMU NS UCODE IMG */
+#define NVGPU_PMU_NS_UCODE_IMAGE        "gpmu_ucode.bin"
+
+/* PMU F/W version */
+#define APP_VERSION_NC_3        22204331
+#define APP_VERSION_NC_2        20429989
+#define APP_VERSION_NC_1        20313802
+#define APP_VERSION_NC_0        20360931
+#define APP_VERSION_GM206       20652057
+#define APP_VERSION_NV_GPU      21307569
+#define APP_VERSION_NV_GPU_1    21308030
+#define APP_VERSION_GM20B_5 20490253
+#define APP_VERSION_GM20B_4 19008461
+#define APP_VERSION_GM20B_3 18935575
+#define APP_VERSION_GM20B_2 18694072
+#define APP_VERSION_GM20B_1 18547257
+#define APP_VERSION_GM20B 17615280
+#define APP_VERSION_3 18357968
+#define APP_VERSION_2 18542378
+#define APP_VERSION_1 17997577 /*Obsolete this once 18357968 gets in*/
+#define APP_VERSION_0 16856675
+
+/* PMU version specific functions */
+static u32 pmu_perfmon_cntr_sz_v0(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_perfmon_counter_v0);
+}
+
+static u32 pmu_perfmon_cntr_sz_v2(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_perfmon_counter_v2);
+}
+
+static void *get_perfmon_cntr_ptr_v2(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->perfmon_counter_v2);
+}
+
+static void *get_perfmon_cntr_ptr_v0(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->perfmon_counter_v0);
+}
+
+static void set_perfmon_cntr_ut_v2(struct nvgpu_pmu *pmu, u16 ut)
+{
+        pmu->perfmon_counter_v2.upper_threshold = ut;
+}
+
+static void set_perfmon_cntr_ut_v0(struct nvgpu_pmu *pmu, u16 ut)
+{
+        pmu->perfmon_counter_v0.upper_threshold = ut;
+}
+
+static void set_perfmon_cntr_lt_v2(struct nvgpu_pmu *pmu, u16 lt)
+{
+        pmu->perfmon_counter_v2.lower_threshold = lt;
+}
+
+static void set_perfmon_cntr_lt_v0(struct nvgpu_pmu *pmu, u16 lt)
+{
+        pmu->perfmon_counter_v0.lower_threshold = lt;
+}
+
+static void set_perfmon_cntr_valid_v2(struct nvgpu_pmu *pmu, u8 valid)
+{
+        pmu->perfmon_counter_v2.valid = valid;
+}
+
+static void set_perfmon_cntr_valid_v0(struct nvgpu_pmu *pmu, u8 valid)
+{
+        pmu->perfmon_counter_v0.valid = valid;
+}
+
+static void set_perfmon_cntr_index_v2(struct nvgpu_pmu *pmu, u8 index)
+{
+        pmu->perfmon_counter_v2.index = index;
+}
+
+static void set_perfmon_cntr_index_v0(struct nvgpu_pmu *pmu, u8 index)
+{
+        pmu->perfmon_counter_v0.index = index;
+}
+
+static void set_perfmon_cntr_group_id_v2(struct nvgpu_pmu *pmu, u8 gid)
+{
+        pmu->perfmon_counter_v2.group_id = gid;
+}
+
+static void set_perfmon_cntr_group_id_v0(struct nvgpu_pmu *pmu, u8 gid)
+{
+        pmu->perfmon_counter_v0.group_id = gid;
+}
+
+static u32 pmu_cmdline_size_v0(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v0);
+}
+
+static u32 pmu_cmdline_size_v1(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v1);
+}
+
+static u32 pmu_cmdline_size_v2(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v2);
+}
+
+static void set_pmu_cmdline_args_cpufreq_v2(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v2.cpu_freq_hz = freq;
+}
+static void set_pmu_cmdline_args_secure_mode_v2(struct nvgpu_pmu *pmu, u32 val)
+{
+        pmu->args_v2.secure_mode = val;
+}
+
+static void set_pmu_cmdline_args_falctracesize_v2(
+                        struct nvgpu_pmu *pmu, u32 size)
+{
+        pmu->args_v2.falc_trace_size = size;
+}
+
+static void set_pmu_cmdline_args_falctracedmabase_v2(struct nvgpu_pmu *pmu)
+{
+        pmu->args_v2.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
+}
+
+static void set_pmu_cmdline_args_falctracedmaidx_v2(
+                        struct nvgpu_pmu *pmu, u32 idx)
+{
+        pmu->args_v2.falc_trace_dma_idx = idx;
+}
+
+
+static void set_pmu_cmdline_args_falctracedmabase_v4(struct nvgpu_pmu *pmu)
+{
+        pmu->args_v4.dma_addr.dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
+        pmu->args_v4.dma_addr.dma_base1 = 0;
+        pmu->args_v4.dma_addr.dma_offset = 0;
+}
+
+static u32 pmu_cmdline_size_v4(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v4);
+}
+
+static void set_pmu_cmdline_args_cpufreq_v4(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v4.cpu_freq_hz = freq;
+}
+static void set_pmu_cmdline_args_secure_mode_v4(struct nvgpu_pmu *pmu, u32 val)
+{
+        pmu->args_v4.secure_mode = val;
+}
+
+static void set_pmu_cmdline_args_falctracesize_v4(
+                        struct nvgpu_pmu *pmu, u32 size)
+{
+        pmu->args_v4.falc_trace_size = size;
+}
+static void set_pmu_cmdline_args_falctracedmaidx_v4(
+                        struct nvgpu_pmu *pmu, u32 idx)
+{
+        pmu->args_v4.falc_trace_dma_idx = idx;
+}
+
+static u32 pmu_cmdline_size_v5(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v5);
+}
+
+static u32 pmu_cmdline_size_v6(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v6);
+}
+
+static void set_pmu_cmdline_args_cpufreq_v5(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v5.cpu_freq_hz = 204000000;
+}
+static void set_pmu_cmdline_args_secure_mode_v5(struct nvgpu_pmu *pmu, u32 val)
+{
+        pmu->args_v5.secure_mode = val;
+}
+
+static void set_pmu_cmdline_args_falctracesize_v5(
+                        struct nvgpu_pmu *pmu, u32 size)
+{
+        /* set by surface describe */
+}
+
+static void set_pmu_cmdline_args_falctracedmabase_v5(struct nvgpu_pmu *pmu)
+{
+        struct gk20a *g = gk20a_from_pmu(pmu);
+
+        nvgpu_pmu_surface_describe(g, &pmu->trace_buf, &pmu->args_v5.trace_buf);
+}
+
+static void set_pmu_cmdline_args_falctracedmaidx_v5(
+                        struct nvgpu_pmu *pmu, u32 idx)
+{
+        /* set by surface describe */
+}
+
+static u32 pmu_cmdline_size_v3(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_cmdline_args_v3);
+}
+
+static void set_pmu_cmdline_args_cpufreq_v3(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v3.cpu_freq_hz = freq;
+}
+static void set_pmu_cmdline_args_secure_mode_v3(struct nvgpu_pmu *pmu, u32 val)
+{
+        pmu->args_v3.secure_mode = val;
+}
+
+static void set_pmu_cmdline_args_falctracesize_v3(
+                        struct nvgpu_pmu *pmu, u32 size)
+{
+        pmu->args_v3.falc_trace_size = size;
+}
+
+static void set_pmu_cmdline_args_falctracedmabase_v3(struct nvgpu_pmu *pmu)
+{
+        pmu->args_v3.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
+}
+
+static void set_pmu_cmdline_args_falctracedmaidx_v3(
+                        struct nvgpu_pmu *pmu, u32 idx)
+{
+        pmu->args_v3.falc_trace_dma_idx = idx;
+}
+
+static void set_pmu_cmdline_args_cpufreq_v1(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v1.cpu_freq_hz = freq;
+}
+static void set_pmu_cmdline_args_secure_mode_v1(struct nvgpu_pmu *pmu, u32 val)
+{
+        pmu->args_v1.secure_mode = val;
+}
+
+static void set_pmu_cmdline_args_falctracesize_v1(
+                        struct nvgpu_pmu *pmu, u32 size)
+{
+        pmu->args_v1.falc_trace_size = size;
+}
+
+static void set_pmu_cmdline_args_falctracedmabase_v1(struct nvgpu_pmu *pmu)
+{
+        pmu->args_v1.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
+}
+
+static void set_pmu_cmdline_args_falctracedmaidx_v1(
+                        struct nvgpu_pmu *pmu, u32 idx)
+{
+        pmu->args_v1.falc_trace_dma_idx = idx;
+}
+
+static void set_pmu_cmdline_args_cpufreq_v0(struct nvgpu_pmu *pmu, u32 freq)
+{
+        pmu->args_v0.cpu_freq_hz = freq;
+}
+
+static void *get_pmu_cmdline_args_ptr_v4(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v4);
+}
+
+static void *get_pmu_cmdline_args_ptr_v3(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v3);
+}
+
+static void *get_pmu_cmdline_args_ptr_v2(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v2);
+}
+
+static void *get_pmu_cmdline_args_ptr_v5(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v5);
+}
+static void *get_pmu_cmdline_args_ptr_v1(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v1);
+}
+
+static void *get_pmu_cmdline_args_ptr_v0(struct nvgpu_pmu *pmu)
+{
+        return (void *)(&pmu->args_v0);
+}
+
+static u32 get_pmu_allocation_size_v3(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_allocation_v3);
+}
+
+static u32 get_pmu_allocation_size_v2(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_allocation_v2);
+}
+
+static u32 get_pmu_allocation_size_v1(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_allocation_v1);
+}
+
+static u32 get_pmu_allocation_size_v0(struct nvgpu_pmu *pmu)
+{
+        return sizeof(struct pmu_allocation_v0);
+}
+
+static void set_pmu_allocation_ptr_v3(struct nvgpu_pmu *pmu,
+        void **pmu_alloc_ptr, void *assign_ptr)
+{
+        struct pmu_allocation_v3 **pmu_a_ptr =
+                (struct pmu_allocation_v3 **)pmu_alloc_ptr;
+
+        *pmu_a_ptr = (struct pmu_allocation_v3 *)assign_ptr;
+}
+
+static void set_pmu_allocation_ptr_v2(struct nvgpu_pmu *pmu,
+        void **pmu_alloc_ptr, void *assign_ptr)
+{
+        struct pmu_allocation_v2 **pmu_a_ptr =
+                (struct pmu_allocation_v2 **)pmu_alloc_ptr;
+
+        *pmu_a_ptr = (struct pmu_allocation_v2 *)assign_ptr;
+}
+
+static void set_pmu_allocation_ptr_v1(struct nvgpu_pmu *pmu,
+        void **pmu_alloc_ptr, void *assign_ptr)
+{
+        struct pmu_allocation_v1 **pmu_a_ptr =
+                (struct pmu_allocation_v1 **)pmu_alloc_ptr;
+
+        *pmu_a_ptr = (struct pmu_allocation_v1 *)assign_ptr;
+}
+
+static void set_pmu_allocation_ptr_v0(struct nvgpu_pmu *pmu,
+        void **pmu_alloc_ptr, void *assign_ptr)
+{
+        struct pmu_allocation_v0 **pmu_a_ptr =
+                (struct pmu_allocation_v0 **)pmu_alloc_ptr;
+
+        *pmu_a_ptr = (struct pmu_allocation_v0 *)assign_ptr;
+}
+
+static void pmu_allocation_set_dmem_size_v3(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u16 size)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.size = size;
+}
+
+static void pmu_allocation_set_dmem_size_v2(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u16 size)
+{
+        struct pmu_allocation_v2 *pmu_a_ptr =
+                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.size = size;
+}
+
+static void pmu_allocation_set_dmem_size_v1(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u16 size)
+{
+        struct pmu_allocation_v1 *pmu_a_ptr =
+                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.size = size;
+}
+
+static void pmu_allocation_set_dmem_size_v0(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u16 size)
+{
+        struct pmu_allocation_v0 *pmu_a_ptr =
+                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.size = size;
+}
+
+static u16 pmu_allocation_get_dmem_size_v3(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.size;
+}
+
+static u16 pmu_allocation_get_dmem_size_v2(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v2 *pmu_a_ptr =
+                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.size;
+}
+
+static u16 pmu_allocation_get_dmem_size_v1(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v1 *pmu_a_ptr =
+                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.size;
+}
+
+static u16 pmu_allocation_get_dmem_size_v0(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v0 *pmu_a_ptr =
+                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.size;
+}
+
+static u32 pmu_allocation_get_dmem_offset_v3(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 pmu_allocation_get_dmem_offset_v2(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v2 *pmu_a_ptr =
+                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 pmu_allocation_get_dmem_offset_v1(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v1 *pmu_a_ptr =
+                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 pmu_allocation_get_dmem_offset_v0(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v0 *pmu_a_ptr =
+                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
+
+        return pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 *pmu_allocation_get_dmem_offset_addr_v3(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        return &pmu_a_ptr->alloc.dmem.offset;
+}
+
+static void *pmu_allocation_get_fb_addr_v3(
+                                struct nvgpu_pmu *pmu, void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                        (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        return (void *)&pmu_a_ptr->alloc.fb;
+}
+
+static u32 pmu_allocation_get_fb_size_v3(
+                                struct nvgpu_pmu *pmu, void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                        (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        return sizeof(pmu_a_ptr->alloc.fb);
+}
+
+static u32 *pmu_allocation_get_dmem_offset_addr_v2(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v2 *pmu_a_ptr =
+                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
+
+        return &pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 *pmu_allocation_get_dmem_offset_addr_v1(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v1 *pmu_a_ptr =
+                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
+
+        return &pmu_a_ptr->alloc.dmem.offset;
+}
+
+static u32 *pmu_allocation_get_dmem_offset_addr_v0(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr)
+{
+        struct pmu_allocation_v0 *pmu_a_ptr =
+                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
+
+        return &pmu_a_ptr->alloc.dmem.offset;
+}
+
+static void pmu_allocation_set_dmem_offset_v3(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u32 offset)
+{
+        struct pmu_allocation_v3 *pmu_a_ptr =
+                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.offset = offset;
+}
+
+static void pmu_allocation_set_dmem_offset_v2(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u32 offset)
+{
+        struct pmu_allocation_v2 *pmu_a_ptr =
+                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.offset = offset;
+}
+
+static void pmu_allocation_set_dmem_offset_v1(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u32 offset)
+{
+        struct pmu_allocation_v1 *pmu_a_ptr =
+                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.offset = offset;
+}
+
+static void pmu_allocation_set_dmem_offset_v0(struct nvgpu_pmu *pmu,
+        void *pmu_alloc_ptr, u32 offset)
+{
+        struct pmu_allocation_v0 *pmu_a_ptr =
+                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
+
+        pmu_a_ptr->alloc.dmem.offset = offset;
+}
+
+static void *get_pmu_msg_pmu_init_msg_ptr_v4(struct pmu_init_msg *init)
+{
+        return (void *)(&(init->pmu_init_v4));
+}
+
+static void *get_pmu_msg_pmu_init_msg_ptr_v3(struct pmu_init_msg *init)
+{
+        return (void *)(&(init->pmu_init_v3));
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_off_v4(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v4 *init =
+                (struct pmu_init_msg_pmu_v4 *)(&init_msg->v4);
+
+        return init->sw_managed_area_offset;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_off_v3(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v3 *init =
+                (struct pmu_init_msg_pmu_v3 *)(&init_msg->v3);
+
+        return init->sw_managed_area_offset;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_size_v4(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v4 *init =
+                (struct pmu_init_msg_pmu_v4 *)(&init_msg->v4);
+
+        return init->sw_managed_area_size;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_size_v3(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v3 *init =
+                (struct pmu_init_msg_pmu_v3 *)(&init_msg->v3);
+
+        return init->sw_managed_area_size;
+}
+
+static void *get_pmu_msg_pmu_init_msg_ptr_v2(struct pmu_init_msg *init)
+{
+        return (void *)(&(init->pmu_init_v2));
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_off_v2(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v2 *init =
+                (struct pmu_init_msg_pmu_v2 *)(&init_msg->v1);
+
+        return init->sw_managed_area_offset;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_size_v2(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v2 *init =
+                (struct pmu_init_msg_pmu_v2 *)(&init_msg->v1);
+
+        return init->sw_managed_area_size;
+}
+
+static void *get_pmu_msg_pmu_init_msg_ptr_v1(struct pmu_init_msg *init)
+{
+        return (void *)(&(init->pmu_init_v1));
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_off_v1(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v1 *init =
+                (struct pmu_init_msg_pmu_v1 *)(&init_msg->v1);
+
+        return init->sw_managed_area_offset;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_size_v1(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v1 *init =
+                (struct pmu_init_msg_pmu_v1 *)(&init_msg->v1);
+
+        return init->sw_managed_area_size;
+}
+
+static void *get_pmu_msg_pmu_init_msg_ptr_v0(struct pmu_init_msg *init)
+{
+        return (void *)(&(init->pmu_init_v0));
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_off_v0(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v0 *init =
+                (struct pmu_init_msg_pmu_v0 *)(&init_msg->v0);
+
+        return init->sw_managed_area_offset;
+}
+
+static u16 get_pmu_init_msg_pmu_sw_mg_size_v0(union pmu_init_msg_pmu *init_msg)
+{
+        struct pmu_init_msg_pmu_v0 *init =
+                (struct pmu_init_msg_pmu_v0 *)(&init_msg->v0);
+
+        return init->sw_managed_area_size;
+}
+
+static u32 get_pmu_perfmon_cmd_start_size_v3(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_start_v3);
+}
+
+static u32 get_pmu_perfmon_cmd_start_size_v2(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_start_v2);
+}
+
+static u32 get_pmu_perfmon_cmd_start_size_v1(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_start_v1);
+}
+
+static u32 get_pmu_perfmon_cmd_start_size_v0(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_start_v0);
+}
+
+static int get_perfmon_cmd_start_offsetofvar_v3(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_start_v3,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_start_offsetofvar_v2(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_start_v2,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_start_offsetofvar_v1(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_start_v1,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_start_offsetofvar_v0(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_start_v0,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static u32 get_pmu_perfmon_cmd_init_size_v3(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_init_v3);
+}
+
+static u32 get_pmu_perfmon_cmd_init_size_v2(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_init_v2);
+}
+
+static u32 get_pmu_perfmon_cmd_init_size_v1(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_init_v1);
+}
+
+static u32 get_pmu_perfmon_cmd_init_size_v0(void)
+{
+        return sizeof(struct pmu_perfmon_cmd_init_v0);
+}
+
+static int get_perfmon_cmd_init_offsetofvar_v3(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_init_v3,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_init_offsetofvar_v2(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_init_v2,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_init_offsetofvar_v1(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_init_v1,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static int get_perfmon_cmd_init_offsetofvar_v0(
+        enum pmu_perfmon_cmd_start_fields field)
+{
+        switch (field) {
+        case COUNTER_ALLOC:
+                return offsetof(struct pmu_perfmon_cmd_init_v0,
+                counter_alloc);
+        default:
+                return -EINVAL;
+        }
+
+        return 0;
+}
+
+static void perfmon_start_set_cmd_type_v3(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
+
+        start->cmd_type = value;
+}
+
+static void perfmon_start_set_cmd_type_v2(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
+
+        start->cmd_type = value;
+}
+
+static void perfmon_start_set_cmd_type_v1(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
+
+        start->cmd_type = value;
+}
+
+static void perfmon_start_set_cmd_type_v0(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
+
+        start->cmd_type = value;
+}
+
+static void perfmon_start_set_group_id_v3(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
+
+        start->group_id = value;
+}
+
+static void perfmon_start_set_group_id_v2(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
+
+        start->group_id = value;
+}
+
+static void perfmon_start_set_group_id_v1(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
+
+        start->group_id = value;
+}
+
+static void perfmon_start_set_group_id_v0(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
+
+        start->group_id = value;
+}
+
+static void perfmon_start_set_state_id_v3(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
+
+        start->state_id = value;
+}
+
+static void perfmon_start_set_state_id_v2(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
+
+        start->state_id = value;
+}
+
+static void perfmon_start_set_state_id_v1(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
+
+        start->state_id = value;
+}
+
+static void perfmon_start_set_state_id_v0(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
+
+        start->state_id = value;
+}
+
+static void perfmon_start_set_flags_v3(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
+
+        start->flags = value;
+}
+
+static void perfmon_start_set_flags_v2(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
+
+        start->flags = value;
+}
+
+static void perfmon_start_set_flags_v1(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
+
+        start->flags = value;
+}
+
+static void perfmon_start_set_flags_v0(struct pmu_perfmon_cmd *pc, u8 value)
+{
+        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
+
+        start->flags = value;
+}
+
+static u8 perfmon_start_get_flags_v3(struct pmu_perfmon_cmd *pc)
+{
+        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
+
+        return start->flags;
+}
+
+static u8 perfmon_start_get_flags_v2(struct pmu_perfmon_cmd *pc)
+{
+        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
+
+        return start->flags;
+}
+
+static u8 perfmon_start_get_flags_v1(struct pmu_perfmon_cmd *pc)
+{
+        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
+
+        return start->flags;
+}
+
+static u8 perfmon_start_get_flags_v0(struct pmu_perfmon_cmd *pc)
+{
+        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
+
+        return start->flags;
+}
+
+static void perfmon_cmd_init_set_sample_buffer_v3(struct pmu_perfmon_cmd *pc,
+        u16 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->sample_buffer = value;
+}
+
+static void perfmon_cmd_init_set_sample_buffer_v2(struct pmu_perfmon_cmd *pc,
+        u16 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->sample_buffer = value;
+}
+
+
+static void perfmon_cmd_init_set_sample_buffer_v1(struct pmu_perfmon_cmd *pc,
+        u16 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->sample_buffer = value;
+}
+
+static void perfmon_cmd_init_set_sample_buffer_v0(struct pmu_perfmon_cmd *pc,
+        u16 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->sample_buffer = value;
+}
+
+static void perfmon_cmd_init_set_dec_cnt_v3(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->to_decrease_count = value;
+}
+
+static void perfmon_cmd_init_set_dec_cnt_v2(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->to_decrease_count = value;
+}
+
+static void perfmon_cmd_init_set_dec_cnt_v1(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->to_decrease_count = value;
+}
+
+static void perfmon_cmd_init_set_dec_cnt_v0(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->to_decrease_count = value;
+}
+
+static void perfmon_cmd_init_set_base_cnt_id_v3(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->base_counter_id = value;
+}
+
+static void perfmon_cmd_init_set_base_cnt_id_v2(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->base_counter_id = value;
+}
+
+static void perfmon_cmd_init_set_base_cnt_id_v1(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->base_counter_id = value;
+}
+
+static void perfmon_cmd_init_set_base_cnt_id_v0(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->base_counter_id = value;
+}
+
+static void perfmon_cmd_init_set_samp_period_us_v3(struct pmu_perfmon_cmd *pc,
+        u32 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->sample_period_us = value;
+}
+
+static void perfmon_cmd_init_set_samp_period_us_v2(struct pmu_perfmon_cmd *pc,
+        u32 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->sample_period_us = value;
+}
+
+static void perfmon_cmd_init_set_samp_period_us_v1(struct pmu_perfmon_cmd *pc,
+        u32 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->sample_period_us = value;
+}
+
+static void perfmon_cmd_init_set_samp_period_us_v0(struct pmu_perfmon_cmd *pc,
+        u32 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->sample_period_us = value;
+}
+
+static void perfmon_cmd_init_set_num_cnt_v3(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->num_counters = value;
+}
+
+static void perfmon_cmd_init_set_num_cnt_v2(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->num_counters = value;
+}
+
+static void perfmon_cmd_init_set_num_cnt_v1(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->num_counters = value;
+}
+
+static void perfmon_cmd_init_set_num_cnt_v0(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->num_counters = value;
+}
+
+static void perfmon_cmd_init_set_mov_avg_v3(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
+
+        init->samples_in_moving_avg = value;
+}
+
+static void perfmon_cmd_init_set_mov_avg_v2(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
+
+        init->samples_in_moving_avg = value;
+}
+
+static void perfmon_cmd_init_set_mov_avg_v1(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
+
+        init->samples_in_moving_avg = value;
+}
+
+static void perfmon_cmd_init_set_mov_avg_v0(struct pmu_perfmon_cmd *pc,
+        u8 value)
+{
+        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
+
+        init->samples_in_moving_avg = value;
+}
+
+static void get_pmu_init_msg_pmu_queue_params_v0(struct pmu_queue *queue,
+        u32 id, void *pmu_init_msg)
+{
+        struct pmu_init_msg_pmu_v0 *init =
+                (struct pmu_init_msg_pmu_v0 *)pmu_init_msg;
+
+        queue->index    = init->queue_info[id].index;
+        queue->offset   = init->queue_info[id].offset;
+        queue->size = init->queue_info[id].size;
+}
+
+static void get_pmu_init_msg_pmu_queue_params_v1(struct pmu_queue *queue,
+        u32 id, void *pmu_init_msg)
+{
+        struct pmu_init_msg_pmu_v1 *init =
+                (struct pmu_init_msg_pmu_v1 *)pmu_init_msg;
+
+        queue->index    = init->queue_info[id].index;
+        queue->offset   = init->queue_info[id].offset;
+        queue->size = init->queue_info[id].size;
+}
+
+static void get_pmu_init_msg_pmu_queue_params_v2(struct pmu_queue *queue,
+        u32 id, void *pmu_init_msg)
+{
+        struct pmu_init_msg_pmu_v2 *init =
+                (struct pmu_init_msg_pmu_v2 *)pmu_init_msg;
+
+        queue->index    = init->queue_info[id].index;
+        queue->offset   = init->queue_info[id].offset;
+        queue->size = init->queue_info[id].size;
+}
+
+static void get_pmu_init_msg_pmu_queue_params_v4(struct pmu_queue *queue,
+        u32 id, void *pmu_init_msg)
+{
+        struct pmu_init_msg_pmu_v4 *init = pmu_init_msg;
+        u32 current_ptr = 0;
+        u8 i;
+        u8 tmp_id = id;
+
+        if (tmp_id == PMU_COMMAND_QUEUE_HPQ)
+                tmp_id = PMU_QUEUE_HPQ_IDX_FOR_V3;
+        else if (tmp_id == PMU_COMMAND_QUEUE_LPQ)
+                tmp_id = PMU_QUEUE_LPQ_IDX_FOR_V3;
+        else if (tmp_id == PMU_MESSAGE_QUEUE)
+                tmp_id = PMU_QUEUE_MSG_IDX_FOR_V3;
+        else
+                return;
+
+        queue->index    = init->queue_index[tmp_id];
+        queue->size = init->queue_size[tmp_id];
+        if (tmp_id != 0) {
+                for (i = 0 ; i < tmp_id; i++)
+                        current_ptr += init->queue_size[i];
+        }
+        queue->offset   = init->queue_offset + current_ptr;
+}
+
+static void get_pmu_init_msg_pmu_queue_params_v3(struct pmu_queue *queue,
+        u32 id, void *pmu_init_msg)
+{
+        struct pmu_init_msg_pmu_v3 *init =
+                (struct pmu_init_msg_pmu_v3 *)pmu_init_msg;
+        u32 current_ptr = 0;
+        u8 i;
+        u8 tmp_id = id;
+
+        if (tmp_id == PMU_COMMAND_QUEUE_HPQ)
+                tmp_id = PMU_QUEUE_HPQ_IDX_FOR_V3;
+        else if (tmp_id == PMU_COMMAND_QUEUE_LPQ)
+                tmp_id = PMU_QUEUE_LPQ_IDX_FOR_V3;
+        else if (tmp_id == PMU_MESSAGE_QUEUE)
+                tmp_id = PMU_QUEUE_MSG_IDX_FOR_V3;
+        else
+                return;
+        queue->index    = init->queue_index[tmp_id];
+        queue->size = init->queue_size[tmp_id];
+        if (tmp_id != 0) {
+                for (i = 0 ; i < tmp_id; i++)
+                        current_ptr += init->queue_size[i];
+        }
+        queue->offset   = init->queue_offset + current_ptr;
+}
+
+static void *get_pmu_sequence_in_alloc_ptr_v3(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->in_v3);
+}
+
+static void *get_pmu_sequence_in_alloc_ptr_v1(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->in_v1);
+}
+
+static void *get_pmu_sequence_in_alloc_ptr_v0(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->in_v0);
+}
+
+static void *get_pmu_sequence_out_alloc_ptr_v3(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->out_v3);
+}
+
+static void *get_pmu_sequence_out_alloc_ptr_v1(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->out_v1);
+}
+
+static void *get_pmu_sequence_out_alloc_ptr_v0(struct pmu_sequence *seq)
+{
+        return (void *)(&seq->out_v0);
+}
+
+static u8 pg_cmd_eng_buf_load_size_v0(struct pmu_pg_cmd *pg)
+{
+        return sizeof(pg->eng_buf_load_v0);
+}
+
+static u8 pg_cmd_eng_buf_load_size_v1(struct pmu_pg_cmd *pg)
+{
+        return sizeof(pg->eng_buf_load_v1);
+}
+
+static u8 pg_cmd_eng_buf_load_size_v2(struct pmu_pg_cmd *pg)
+{
+        return sizeof(pg->eng_buf_load_v2);
+}
+
+static void pg_cmd_eng_buf_load_set_cmd_type_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.cmd_type = value;
+}
+
+static void pg_cmd_eng_buf_load_set_cmd_type_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.cmd_type = value;
+}
+
+static void pg_cmd_eng_buf_load_set_cmd_type_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.cmd_type = value;
+}
+
+static void pg_cmd_eng_buf_load_set_engine_id_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.engine_id = value;
+}
+static void pg_cmd_eng_buf_load_set_engine_id_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.engine_id = value;
+}
+static void pg_cmd_eng_buf_load_set_engine_id_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.engine_id = value;
+}
+static void pg_cmd_eng_buf_load_set_buf_idx_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.buf_idx = value;
+}
+static void pg_cmd_eng_buf_load_set_buf_idx_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.buf_idx = value;
+}
+static void pg_cmd_eng_buf_load_set_buf_idx_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.buf_idx = value;
+}
+
+static void pg_cmd_eng_buf_load_set_pad_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.pad = value;
+}
+static void pg_cmd_eng_buf_load_set_pad_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.pad = value;
+}
+static void pg_cmd_eng_buf_load_set_pad_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.pad = value;
+}
+
+static void pg_cmd_eng_buf_load_set_buf_size_v0(struct pmu_pg_cmd *pg,
+        u16 value)
+{
+        pg->eng_buf_load_v0.buf_size = value;
+}
+static void pg_cmd_eng_buf_load_set_buf_size_v1(struct pmu_pg_cmd *pg,
+        u16 value)
+{
+        pg->eng_buf_load_v1.dma_desc.dma_size = value;
+}
+static void pg_cmd_eng_buf_load_set_buf_size_v2(struct pmu_pg_cmd *pg,
+        u16 value)
+{
+        pg->eng_buf_load_v2.dma_desc.params = value;
+}
+
+static void pg_cmd_eng_buf_load_set_dma_base_v0(struct pmu_pg_cmd *pg,
+        u32 value)
+{
+        pg->eng_buf_load_v0.dma_base = (value >> 8);
+}
+static void pg_cmd_eng_buf_load_set_dma_base_v1(struct pmu_pg_cmd *pg,
+        u32 value)
+{
+        pg->eng_buf_load_v1.dma_desc.dma_addr.lo |= u64_lo32(value);
+        pg->eng_buf_load_v1.dma_desc.dma_addr.hi |= u64_hi32(value);
+}
+static void pg_cmd_eng_buf_load_set_dma_base_v2(struct pmu_pg_cmd *pg,
+        u32 value)
+{
+        pg->eng_buf_load_v2.dma_desc.address.lo = u64_lo32(value);
+        pg->eng_buf_load_v2.dma_desc.address.hi = u64_lo32(value);
+}
+
+static void pg_cmd_eng_buf_load_set_dma_offset_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.dma_offset = value;
+}
+static void pg_cmd_eng_buf_load_set_dma_offset_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.dma_desc.dma_addr.lo |= value;
+}
+static void pg_cmd_eng_buf_load_set_dma_offset_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.dma_desc.address.lo |= u64_lo32(value);
+        pg->eng_buf_load_v2.dma_desc.address.hi |= u64_lo32(value);
+}
+
+static void pg_cmd_eng_buf_load_set_dma_idx_v0(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v0.dma_idx = value;
+}
+
+static void pg_cmd_eng_buf_load_set_dma_idx_v1(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v1.dma_desc.dma_idx = value;
+}
+
+static void pg_cmd_eng_buf_load_set_dma_idx_v2(struct pmu_pg_cmd *pg,
+        u8 value)
+{
+        pg->eng_buf_load_v2.dma_desc.params |= (value << 24);
+}
+
+static int nvgpu_init_pmu_fw_ver_ops(struct nvgpu_pmu *pmu)
+{
+        struct gk20a *g = gk20a_from_pmu(pmu);
+        struct pmu_v *pv = &g->ops.pmu_ver;
+        int err = 0;
+
+        nvgpu_log_fn(g, " ");
+
+        switch (pmu->desc->app_version) {
+        case APP_VERSION_NC_2:
+        case APP_VERSION_NC_1:
+        case APP_VERSION_NC_0:
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
+                                pg_cmd_eng_buf_load_size_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
+                                pg_cmd_eng_buf_load_set_cmd_type_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v1;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v1;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v4;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v4;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v4;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v4;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v4;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v4;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v4;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v2;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v2;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v2;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v2;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v2;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v2;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v2;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v1;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v2;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v2;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v2;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v2;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v2;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v2;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v2;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v2;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v2;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v2;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v1;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v1;
+                break;
+        case APP_VERSION_NC_3:
+                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 =
+                                pg_cmd_eng_buf_load_set_cmd_type_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = false;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v6;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v5;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v5;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v3;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v3;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v3;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v3;
+                g->ops.pmu_ver.pmu_allocation_get_fb_addr =
+                                pmu_allocation_get_fb_addr_v3;
+                g->ops.pmu_ver.pmu_allocation_get_fb_size =
+                                pmu_allocation_get_fb_size_v3;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v4;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v4;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v4;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v4;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v3;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v3;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v3;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v3;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v3;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v3;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v3;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v3;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v3;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v3;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v3;
+                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 =
+                                pg_cmd_eng_buf_load_set_cmd_type_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v2;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v5;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v5;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v5;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v3;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v3;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v3;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v3;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v3;
+                g->ops.pmu_ver.pmu_allocation_get_fb_addr =
+                                pmu_allocation_get_fb_addr_v3;
+                g->ops.pmu_ver.pmu_allocation_get_fb_size =
+                                pmu_allocation_get_fb_size_v3;
+                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 =
+                                        get_pmu_msg_pmu_init_msg_ptr_v2;
+                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                                        get_pmu_init_msg_pmu_sw_mg_off_v2;
+                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                                        get_pmu_init_msg_pmu_sw_mg_size_v2;
+                } else {
+                        g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                                get_pmu_init_msg_pmu_queue_params_v3;
+                        g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                                get_pmu_msg_pmu_init_msg_ptr_v3;
+                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                                get_pmu_init_msg_pmu_sw_mg_off_v3;
+                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                                get_pmu_init_msg_pmu_sw_mg_size_v3;
+                }
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v3;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v3;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v3;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v3;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v3;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v3;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v3;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v3;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v3;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v3;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v3;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v3;
+                break;
+        case APP_VERSION_GM20B_5:
+        case APP_VERSION_GM20B_4:
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
+                                pg_cmd_eng_buf_load_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
+                                pg_cmd_eng_buf_load_set_cmd_type_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v0;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v3;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v3;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v3;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v3;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v3;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v3;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v3;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v1;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v1;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v1;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v1;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v1;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v1;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v1;
+                break;
+        case APP_VERSION_GM20B_3:
+        case APP_VERSION_GM20B_2:
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
+                                pg_cmd_eng_buf_load_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
+                                pg_cmd_eng_buf_load_set_cmd_type_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v0;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v2;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v2;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v2;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v2;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v2;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v2;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v2;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v2;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v2;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v1;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v1;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v1;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v1;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v1;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v1;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v1;
+                break;
+        case APP_VERSION_GM20B_1:
+        case APP_VERSION_GM20B:
+        case APP_VERSION_1:
+        case APP_VERSION_2:
+        case APP_VERSION_3:
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
+                                pg_cmd_eng_buf_load_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
+                                pg_cmd_eng_buf_load_set_cmd_type_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v0;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v0;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v0;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v1;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v1;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        set_pmu_cmdline_args_secure_mode_v1;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
+                        set_pmu_cmdline_args_falctracesize_v1;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
+                        set_pmu_cmdline_args_falctracedmabase_v1;
+                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
+                        set_pmu_cmdline_args_falctracedmaidx_v1;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v1;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v1;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v1;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v1;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v1;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v1;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v1;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v1;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v1;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v1;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v1;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v1;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v1;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v1;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v1;
+                break;
+        case APP_VERSION_0:
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
+                                pg_cmd_eng_buf_load_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
+                                pg_cmd_eng_buf_load_set_cmd_type_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
+                                pg_cmd_eng_buf_load_set_engine_id_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
+                                pg_cmd_eng_buf_load_set_buf_idx_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
+                                pg_cmd_eng_buf_load_set_pad_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
+                                pg_cmd_eng_buf_load_set_buf_size_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
+                                pg_cmd_eng_buf_load_set_dma_base_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
+                                pg_cmd_eng_buf_load_set_dma_offset_v0;
+                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
+                                pg_cmd_eng_buf_load_set_dma_idx_v0;
+                g->ops.pmu_ver.cmd_id_zbc_table_update = 14;
+                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
+                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_valid =
+                        set_perfmon_cntr_valid_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_index =
+                        set_perfmon_cntr_index_v0;
+                g->ops.pmu_ver.set_perfmon_cntr_group_id =
+                        set_perfmon_cntr_group_id_v0;
+                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v0;
+                g->ops.pmu_ver.get_pmu_cmdline_args_size =
+                        pmu_cmdline_size_v0;
+                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
+                        set_pmu_cmdline_args_cpufreq_v0;
+                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
+                        NULL;
+                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
+                        get_pmu_cmdline_args_ptr_v0;
+                g->ops.pmu_ver.get_pmu_allocation_struct_size =
+                        get_pmu_allocation_size_v0;
+                g->ops.pmu_ver.set_pmu_allocation_ptr =
+                        set_pmu_allocation_ptr_v0;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
+                        pmu_allocation_set_dmem_size_v0;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
+                        pmu_allocation_get_dmem_size_v0;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
+                        pmu_allocation_get_dmem_offset_v0;
+                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
+                        pmu_allocation_get_dmem_offset_addr_v0;
+                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
+                        pmu_allocation_set_dmem_offset_v0;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
+                        get_pmu_init_msg_pmu_queue_params_v0;
+                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
+                        get_pmu_msg_pmu_init_msg_ptr_v0;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
+                        get_pmu_init_msg_pmu_sw_mg_off_v0;
+                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
+                        get_pmu_init_msg_pmu_sw_mg_size_v0;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
+                        get_pmu_perfmon_cmd_start_size_v0;
+                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
+                        get_perfmon_cmd_start_offsetofvar_v0;
+                g->ops.pmu_ver.perfmon_start_set_cmd_type =
+                        perfmon_start_set_cmd_type_v0;
+                g->ops.pmu_ver.perfmon_start_set_group_id =
+                        perfmon_start_set_group_id_v0;
+                g->ops.pmu_ver.perfmon_start_set_state_id =
+                        perfmon_start_set_state_id_v0;
+                g->ops.pmu_ver.perfmon_start_set_flags =
+                        perfmon_start_set_flags_v0;
+                g->ops.pmu_ver.perfmon_start_get_flags =
+                        perfmon_start_get_flags_v0;
+                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
+                        get_pmu_perfmon_cmd_init_size_v0;
+                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
+                        get_perfmon_cmd_init_offsetofvar_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
+                        perfmon_cmd_init_set_sample_buffer_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
+                        perfmon_cmd_init_set_dec_cnt_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
+                        perfmon_cmd_init_set_base_cnt_id_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
+                        perfmon_cmd_init_set_samp_period_us_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
+                        perfmon_cmd_init_set_num_cnt_v0;
+                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
+                        perfmon_cmd_init_set_mov_avg_v0;
+                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
+                        get_pmu_sequence_in_alloc_ptr_v0;
+                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
+                        get_pmu_sequence_out_alloc_ptr_v0;
+                break;
+        default:
+                nvgpu_err(g, "PMU code version not supported version: %d\n",
+                        pmu->desc->app_version);
+                err = -EINVAL;
+        }
+        pv->set_perfmon_cntr_index(pmu, 3); /* GR & CE2 */
+        pv->set_perfmon_cntr_group_id(pmu, PMU_DOMAIN_GROUP_PSTATE);
+
+        return err;
+}
+
+static void nvgpu_remove_pmu_support(struct nvgpu_pmu *pmu)
+{
+        struct gk20a *g = gk20a_from_pmu(pmu);
+
+        nvgpu_log_fn(g, " ");
+
+        if (nvgpu_alloc_initialized(&pmu->dmem))
+                nvgpu_alloc_destroy(&pmu->dmem);
+
+        nvgpu_release_firmware(g, pmu->fw);
+
+        nvgpu_mutex_destroy(&pmu->elpg_mutex);
+        nvgpu_mutex_destroy(&pmu->pg_mutex);
+        nvgpu_mutex_destroy(&pmu->isr_mutex);
+        nvgpu_mutex_destroy(&pmu->pmu_copy_lock);
+        nvgpu_mutex_destroy(&pmu->pmu_seq_lock);
+}
+
+int nvgpu_init_pmu_fw_support(struct nvgpu_pmu *pmu)
+{
+        struct gk20a *g = gk20a_from_pmu(pmu);
+        int err = 0;
+
+        nvgpu_log_fn(g, " ");
+
+        err = nvgpu_mutex_init(&pmu->elpg_mutex);
+        if (err)
+                return err;
+
+        err = nvgpu_mutex_init(&pmu->pg_mutex);
+        if (err)
+                goto fail_elpg;
+
+        err = nvgpu_mutex_init(&pmu->isr_mutex);
+        if (err)
+                goto fail_pg;
+
+        err = nvgpu_mutex_init(&pmu->pmu_copy_lock);
+        if (err)
+                goto fail_isr;
+
+        err = nvgpu_mutex_init(&pmu->pmu_seq_lock);
+        if (err)
+                goto fail_pmu_copy;
+
+        pmu->remove_support = nvgpu_remove_pmu_support;
+
+        err = nvgpu_init_pmu_fw_ver_ops(pmu);
+        if (err)
+                goto fail_pmu_seq;
+
+        goto exit;
+
+fail_pmu_seq:
+        nvgpu_mutex_destroy(&pmu->pmu_seq_lock);
+fail_pmu_copy:
+        nvgpu_mutex_destroy(&pmu->pmu_copy_lock);
+fail_isr:
+        nvgpu_mutex_destroy(&pmu->isr_mutex);
+fail_pg:
+        nvgpu_mutex_destroy(&pmu->pg_mutex);
+fail_elpg:
+        nvgpu_mutex_destroy(&pmu->elpg_mutex);
+exit:
+        return err;
+}
+
+int nvgpu_pmu_prepare_ns_ucode_blob(struct gk20a *g)
+{
+        struct nvgpu_pmu *pmu = &g->pmu;
+        int err = 0;
+        struct mm_gk20a *mm = &g->mm;
+        struct vm_gk20a *vm = mm->pmu.vm;
+
+        nvgpu_log_fn(g, " ");
+
+        if (pmu->fw)
+                return nvgpu_init_pmu_fw_support(pmu);
+
+        pmu->fw = nvgpu_request_firmware(g, NVGPU_PMU_NS_UCODE_IMAGE, 0);
+        if (!pmu->fw) {
+                nvgpu_err(g, "failed to load pmu ucode!!");
+                return err;
+        }
+
+        nvgpu_log_fn(g, "firmware loaded");
+
+        pmu->desc = (struct pmu_ucode_desc *)pmu->fw->data;
+        pmu->ucode_image = (u32 *)((u8 *)pmu->desc +
+                        pmu->desc->descriptor_size);
+
+        err = nvgpu_dma_alloc_map_sys(vm, GK20A_PMU_UCODE_SIZE_MAX,
+                        &pmu->ucode);
+        if (err)
+                goto err_release_fw;
+
+        nvgpu_mem_wr_n(g, &pmu->ucode, 0, pmu->ucode_image,
+                        pmu->desc->app_start_offset + pmu->desc->app_size);
+
+        return nvgpu_init_pmu_fw_support(pmu);
+
+ err_release_fw:
+        nvgpu_release_firmware(g, pmu->fw);
+        pmu->fw = NULL;
+
+        return err;
+}
+
diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index a9457330..247b38a5 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -35,249 +35,20 @@
 #include "nvgpu_gpuid_t19x.h"
 #endif
 
-#define GK20A_PMU_UCODE_IMAGE   "gpmu_ucode.bin"
+#define gk20a_dbg_pmu(fmt, arg...) \
+        gk20a_dbg(gpu_dbg_pmu, fmt, ##arg)
 
 #define PMU_MEM_SCRUBBING_TIMEOUT_MAX 1000
 #define PMU_MEM_SCRUBBING_TIMEOUT_DEFAULT 10
 
-#define gk20a_dbg_pmu(fmt, arg...) \
-        gk20a_dbg(gpu_dbg_pmu, fmt, ##arg)
-
 static void ap_callback_init_and_enable_ctrl(
                 struct gk20a *g, struct pmu_msg *msg,
                 void *param, u32 seq_desc, u32 status);
 
-static u32 pmu_perfmon_cntr_sz_v0(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_perfmon_counter_v0);
-}
-
-static u32 pmu_perfmon_cntr_sz_v2(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_perfmon_counter_v2);
-}
-
-static void *get_perfmon_cntr_ptr_v2(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->perfmon_counter_v2);
-}
-
-static void *get_perfmon_cntr_ptr_v0(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->perfmon_counter_v0);
-}
-
-static void set_perfmon_cntr_ut_v2(struct nvgpu_pmu *pmu, u16 ut)
-{
-        pmu->perfmon_counter_v2.upper_threshold = ut;
-}
-
-static void set_perfmon_cntr_ut_v0(struct nvgpu_pmu *pmu, u16 ut)
-{
-        pmu->perfmon_counter_v0.upper_threshold = ut;
-}
-
-static void set_perfmon_cntr_lt_v2(struct nvgpu_pmu *pmu, u16 lt)
-{
-        pmu->perfmon_counter_v2.lower_threshold = lt;
-}
-
-static void set_perfmon_cntr_lt_v0(struct nvgpu_pmu *pmu, u16 lt)
-{
-        pmu->perfmon_counter_v0.lower_threshold = lt;
-}
-
-static void set_perfmon_cntr_valid_v2(struct nvgpu_pmu *pmu, u8 valid)
-{
-        pmu->perfmon_counter_v2.valid = valid;
-}
-
-static void set_perfmon_cntr_valid_v0(struct nvgpu_pmu *pmu, u8 valid)
-{
-        pmu->perfmon_counter_v0.valid = valid;
-}
-
-static void set_perfmon_cntr_index_v2(struct nvgpu_pmu *pmu, u8 index)
-{
-        pmu->perfmon_counter_v2.index = index;
-}
-
-static void set_perfmon_cntr_index_v0(struct nvgpu_pmu *pmu, u8 index)
-{
-        pmu->perfmon_counter_v0.index = index;
-}
-
-static void set_perfmon_cntr_group_id_v2(struct nvgpu_pmu *pmu, u8 gid)
-{
-        pmu->perfmon_counter_v2.group_id = gid;
-}
-
-static void set_perfmon_cntr_group_id_v0(struct nvgpu_pmu *pmu, u8 gid)
-{
-        pmu->perfmon_counter_v0.group_id = gid;
-}
-
-static u32 pmu_cmdline_size_v0(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v0);
-}
-
-static u32 pmu_cmdline_size_v1(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v1);
-}
-
-static u32 pmu_cmdline_size_v2(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v2);
-}
-
-static void set_pmu_cmdline_args_cpufreq_v2(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v2.cpu_freq_hz = freq;
-}
-static void set_pmu_cmdline_args_secure_mode_v2(struct nvgpu_pmu *pmu, u32 val)
-{
-        pmu->args_v2.secure_mode = val;
-}
-
-static void set_pmu_cmdline_args_falctracesize_v2(
-                        struct nvgpu_pmu *pmu, u32 size)
-{
-        pmu->args_v2.falc_trace_size = size;
-}
-
-static void set_pmu_cmdline_args_falctracedmabase_v2(struct nvgpu_pmu *pmu)
-{
-        pmu->args_v2.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
-}
-
-static void set_pmu_cmdline_args_falctracedmaidx_v2(
-                        struct nvgpu_pmu *pmu, u32 idx)
-{
-        pmu->args_v2.falc_trace_dma_idx = idx;
-}
-
-
-static void set_pmu_cmdline_args_falctracedmabase_v4(struct nvgpu_pmu *pmu)
-{
-        pmu->args_v4.dma_addr.dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
-        pmu->args_v4.dma_addr.dma_base1 = 0;
-        pmu->args_v4.dma_addr.dma_offset = 0;
-}
-
-static u32 pmu_cmdline_size_v4(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v4);
-}
-
-static void set_pmu_cmdline_args_cpufreq_v4(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v4.cpu_freq_hz = freq;
-}
-static void set_pmu_cmdline_args_secure_mode_v4(struct nvgpu_pmu *pmu, u32 val)
-{
-        pmu->args_v4.secure_mode = val;
-}
-
-static void set_pmu_cmdline_args_falctracesize_v4(
-                        struct nvgpu_pmu *pmu, u32 size)
-{
-        pmu->args_v4.falc_trace_size = size;
-}
-static void set_pmu_cmdline_args_falctracedmaidx_v4(
-                        struct nvgpu_pmu *pmu, u32 idx)
-{
-        pmu->args_v4.falc_trace_dma_idx = idx;
-}
-
-static u32 pmu_cmdline_size_v5(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v5);
-}
-
-static u32 pmu_cmdline_size_v6(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v6);
-}
-
-static void set_pmu_cmdline_args_cpufreq_v5(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v5.cpu_freq_hz = 204000000;
-}
-static void set_pmu_cmdline_args_secure_mode_v5(struct nvgpu_pmu *pmu, u32 val)
-{
-        pmu->args_v5.secure_mode = val;
-}
-
-static void set_pmu_cmdline_args_falctracesize_v5(
-                        struct nvgpu_pmu *pmu, u32 size)
-{
-        /* set by surface describe */
-}
-
-static void set_pmu_cmdline_args_falctracedmabase_v5(struct nvgpu_pmu *pmu)
-{
-        struct gk20a *g = gk20a_from_pmu(pmu);
-
-        nvgpu_pmu_surface_describe(g, &pmu->trace_buf, &pmu->args_v5.trace_buf);
-}
-
-static void set_pmu_cmdline_args_falctracedmaidx_v5(
-                        struct nvgpu_pmu *pmu, u32 idx)
-{
-        /* set by surface describe */
-}
-
-static u32 pmu_cmdline_size_v3(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_cmdline_args_v3);
-}
-
-static void set_pmu_cmdline_args_cpufreq_v3(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v3.cpu_freq_hz = freq;
-}
-static void set_pmu_cmdline_args_secure_mode_v3(struct nvgpu_pmu *pmu, u32 val)
-{
-        pmu->args_v3.secure_mode = val;
-}
-
-static void set_pmu_cmdline_args_falctracesize_v3(
-                        struct nvgpu_pmu *pmu, u32 size)
-{
-        pmu->args_v3.falc_trace_size = size;
-}
-
-static void set_pmu_cmdline_args_falctracedmabase_v3(struct nvgpu_pmu *pmu)
-{
-        pmu->args_v3.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
-}
-
-static void set_pmu_cmdline_args_falctracedmaidx_v3(
-                        struct nvgpu_pmu *pmu, u32 idx)
-{
-        pmu->args_v3.falc_trace_dma_idx = idx;
-}
-
-static void set_pmu_cmdline_args_cpufreq_v1(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v1.cpu_freq_hz = freq;
-}
-static void set_pmu_cmdline_args_secure_mode_v1(struct nvgpu_pmu *pmu, u32 val)
-{
-        pmu->args_v1.secure_mode = val;
-}
-
-static void set_pmu_cmdline_args_falctracesize_v1(
-                        struct nvgpu_pmu *pmu, u32 size)
-{
-        pmu->args_v1.falc_trace_size = size;
-}
-
 bool nvgpu_find_hex_in_string(char *strings, struct gk20a *g, u32 *hex_pos)
 {
         u32 i = 0, j = strlen(strings);
+
         for (; i < j; i++) {
                 if (strings[i] == '%')
                         if (strings[i + 1] == 'x' || strings[i + 1] == 'X') {
@@ -330,1885 +101,13 @@ static void printtrace(struct nvgpu_pmu *pmu)
                         l++;
                         m += k + 2;
                 }
+
                 scnprintf((buf + count), 0x40, "%s", (trace+i+20+m));
                 nvgpu_err(g, "%s", buf);
         }
         nvgpu_kfree(g, tracebuffer);
 }
 
-static void set_pmu_cmdline_args_falctracedmabase_v1(struct nvgpu_pmu *pmu)
-{
-        pmu->args_v1.falc_trace_dma_base = ((u32)pmu->trace_buf.gpu_va)/0x100;
-}
-
-static void set_pmu_cmdline_args_falctracedmaidx_v1(
-                        struct nvgpu_pmu *pmu, u32 idx)
-{
-        pmu->args_v1.falc_trace_dma_idx = idx;
-}
-
-static void set_pmu_cmdline_args_cpufreq_v0(struct nvgpu_pmu *pmu, u32 freq)
-{
-        pmu->args_v0.cpu_freq_hz = freq;
-}
-
-static void *get_pmu_cmdline_args_ptr_v4(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v4);
-}
-
-static void *get_pmu_cmdline_args_ptr_v3(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v3);
-}
-
-static void *get_pmu_cmdline_args_ptr_v2(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v2);
-}
-
-static void *get_pmu_cmdline_args_ptr_v5(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v5);
-}
-static void *get_pmu_cmdline_args_ptr_v1(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v1);
-}
-
-static void *get_pmu_cmdline_args_ptr_v0(struct nvgpu_pmu *pmu)
-{
-        return (void *)(&pmu->args_v0);
-}
-
-static u32 get_pmu_allocation_size_v3(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_allocation_v3);
-}
-
-static u32 get_pmu_allocation_size_v2(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_allocation_v2);
-}
-
-static u32 get_pmu_allocation_size_v1(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_allocation_v1);
-}
-
-static u32 get_pmu_allocation_size_v0(struct nvgpu_pmu *pmu)
-{
-        return sizeof(struct pmu_allocation_v0);
-}
-
-static void set_pmu_allocation_ptr_v3(struct nvgpu_pmu *pmu,
-        void **pmu_alloc_ptr, void *assign_ptr)
-{
-        struct pmu_allocation_v3 **pmu_a_ptr =
-                (struct pmu_allocation_v3 **)pmu_alloc_ptr;
-        *pmu_a_ptr = (struct pmu_allocation_v3 *)assign_ptr;
-}
-
-static void set_pmu_allocation_ptr_v2(struct nvgpu_pmu *pmu,
-        void **pmu_alloc_ptr, void *assign_ptr)
-{
-        struct pmu_allocation_v2 **pmu_a_ptr =
-                (struct pmu_allocation_v2 **)pmu_alloc_ptr;
-        *pmu_a_ptr = (struct pmu_allocation_v2 *)assign_ptr;
-}
-
-static void set_pmu_allocation_ptr_v1(struct nvgpu_pmu *pmu,
-        void **pmu_alloc_ptr, void *assign_ptr)
-{
-        struct pmu_allocation_v1 **pmu_a_ptr =
-                (struct pmu_allocation_v1 **)pmu_alloc_ptr;
-        *pmu_a_ptr = (struct pmu_allocation_v1 *)assign_ptr;
-}
-
-static void set_pmu_allocation_ptr_v0(struct nvgpu_pmu *pmu,
-        void **pmu_alloc_ptr, void *assign_ptr)
-{
-        struct pmu_allocation_v0 **pmu_a_ptr =
-                (struct pmu_allocation_v0 **)pmu_alloc_ptr;
-        *pmu_a_ptr = (struct pmu_allocation_v0 *)assign_ptr;
-}
-
-static void pmu_allocation_set_dmem_size_v3(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u16 size)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.size = size;
-}
-
-static void pmu_allocation_set_dmem_size_v2(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u16 size)
-{
-        struct pmu_allocation_v2 *pmu_a_ptr =
-                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.size = size;
-}
-
-static void pmu_allocation_set_dmem_size_v1(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u16 size)
-{
-        struct pmu_allocation_v1 *pmu_a_ptr =
-                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.size = size;
-}
-
-static void pmu_allocation_set_dmem_size_v0(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u16 size)
-{
-        struct pmu_allocation_v0 *pmu_a_ptr =
-                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.size = size;
-}
-
-static u16 pmu_allocation_get_dmem_size_v3(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.size;
-}
-
-static u16 pmu_allocation_get_dmem_size_v2(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v2 *pmu_a_ptr =
-                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.size;
-}
-
-static u16 pmu_allocation_get_dmem_size_v1(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v1 *pmu_a_ptr =
-                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.size;
-}
-
-static u16 pmu_allocation_get_dmem_size_v0(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v0 *pmu_a_ptr =
-                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.size;
-}
-
-static u32 pmu_allocation_get_dmem_offset_v3(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 pmu_allocation_get_dmem_offset_v2(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v2 *pmu_a_ptr =
-                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 pmu_allocation_get_dmem_offset_v1(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v1 *pmu_a_ptr =
-                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 pmu_allocation_get_dmem_offset_v0(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v0 *pmu_a_ptr =
-                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
-        return pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 *pmu_allocation_get_dmem_offset_addr_v3(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        return &pmu_a_ptr->alloc.dmem.offset;
-}
-
-static void *pmu_allocation_get_fb_addr_v3(
-                                struct nvgpu_pmu *pmu, void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                        (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        return (void *)&pmu_a_ptr->alloc.fb;
-}
-
-static u32 pmu_allocation_get_fb_size_v3(
-                                struct nvgpu_pmu *pmu, void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                        (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        return sizeof(pmu_a_ptr->alloc.fb);
-}
-
-static u32 *pmu_allocation_get_dmem_offset_addr_v2(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v2 *pmu_a_ptr =
-                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
-        return &pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 *pmu_allocation_get_dmem_offset_addr_v1(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v1 *pmu_a_ptr =
-                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
-        return &pmu_a_ptr->alloc.dmem.offset;
-}
-
-static u32 *pmu_allocation_get_dmem_offset_addr_v0(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr)
-{
-        struct pmu_allocation_v0 *pmu_a_ptr =
-                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
-        return &pmu_a_ptr->alloc.dmem.offset;
-}
-
-static void pmu_allocation_set_dmem_offset_v3(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u32 offset)
-{
-        struct pmu_allocation_v3 *pmu_a_ptr =
-                (struct pmu_allocation_v3 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.offset = offset;
-}
-
-static void pmu_allocation_set_dmem_offset_v2(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u32 offset)
-{
-        struct pmu_allocation_v2 *pmu_a_ptr =
-                (struct pmu_allocation_v2 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.offset = offset;
-}
-
-static void pmu_allocation_set_dmem_offset_v1(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u32 offset)
-{
-        struct pmu_allocation_v1 *pmu_a_ptr =
-                (struct pmu_allocation_v1 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.offset = offset;
-}
-
-static void pmu_allocation_set_dmem_offset_v0(struct nvgpu_pmu *pmu,
-        void *pmu_alloc_ptr, u32 offset)
-{
-        struct pmu_allocation_v0 *pmu_a_ptr =
-                (struct pmu_allocation_v0 *)pmu_alloc_ptr;
-        pmu_a_ptr->alloc.dmem.offset = offset;
-}
-
-static void *get_pmu_msg_pmu_init_msg_ptr_v4(struct pmu_init_msg *init)
-{
-        return (void *)(&(init->pmu_init_v4));
-}
-
-static void *get_pmu_msg_pmu_init_msg_ptr_v3(struct pmu_init_msg *init)
-{
-        return (void *)(&(init->pmu_init_v3));
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_off_v4(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v4 *init =
-                (struct pmu_init_msg_pmu_v4 *)(&init_msg->v4);
-
-        return init->sw_managed_area_offset;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_off_v3(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v3 *init =
-                (struct pmu_init_msg_pmu_v3 *)(&init_msg->v3);
-
-        return init->sw_managed_area_offset;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_size_v4(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v4 *init =
-                (struct pmu_init_msg_pmu_v4 *)(&init_msg->v4);
-
-        return init->sw_managed_area_size;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_size_v3(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v3 *init =
-                (struct pmu_init_msg_pmu_v3 *)(&init_msg->v3);
-
-        return init->sw_managed_area_size;
-}
-
-static void *get_pmu_msg_pmu_init_msg_ptr_v2(struct pmu_init_msg *init)
-{
-        return (void *)(&(init->pmu_init_v2));
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_off_v2(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v2 *init =
-                (struct pmu_init_msg_pmu_v2 *)(&init_msg->v1);
-        return init->sw_managed_area_offset;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_size_v2(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v2 *init =
-                (struct pmu_init_msg_pmu_v2 *)(&init_msg->v1);
-        return init->sw_managed_area_size;
-}
-
-static void *get_pmu_msg_pmu_init_msg_ptr_v1(struct pmu_init_msg *init)
-{
-        return (void *)(&(init->pmu_init_v1));
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_off_v1(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v1 *init =
-                (struct pmu_init_msg_pmu_v1 *)(&init_msg->v1);
-        return init->sw_managed_area_offset;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_size_v1(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v1 *init =
-                (struct pmu_init_msg_pmu_v1 *)(&init_msg->v1);
-        return init->sw_managed_area_size;
-}
-
-static void *get_pmu_msg_pmu_init_msg_ptr_v0(struct pmu_init_msg *init)
-{
-        return (void *)(&(init->pmu_init_v0));
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_off_v0(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v0 *init =
-                (struct pmu_init_msg_pmu_v0 *)(&init_msg->v0);
-        return init->sw_managed_area_offset;
-}
-
-static u16 get_pmu_init_msg_pmu_sw_mg_size_v0(union pmu_init_msg_pmu *init_msg)
-{
-        struct pmu_init_msg_pmu_v0 *init =
-                (struct pmu_init_msg_pmu_v0 *)(&init_msg->v0);
-        return init->sw_managed_area_size;
-}
-
-static u32 get_pmu_perfmon_cmd_start_size_v3(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_start_v3);
-}
-
-static u32 get_pmu_perfmon_cmd_start_size_v2(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_start_v2);
-}
-
-static u32 get_pmu_perfmon_cmd_start_size_v1(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_start_v1);
-}
-
-static u32 get_pmu_perfmon_cmd_start_size_v0(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_start_v0);
-}
-
-static int get_perfmon_cmd_start_offsetofvar_v3(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_start_v3,
-                counter_alloc);
-        default:
-                return -EINVAL;
-        }
-
-        return 0;
-}
-
-static int get_perfmon_cmd_start_offsetofvar_v2(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_start_v2,
-                counter_alloc);
-        default:
-                return -EINVAL;
-        }
-
-        return 0;
-}
-
-static int get_perfmon_cmd_start_offsetofvar_v1(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_start_v1,
-                counter_alloc);
-        default:
-                return -EINVAL;
-        }
-
-        return 0;
-}
-
-static int get_perfmon_cmd_start_offsetofvar_v0(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_start_v0,
-                counter_alloc);
-        default:
-                return -EINVAL;
-                break;
-        }
-        return 0;
-}
-
-static u32 get_pmu_perfmon_cmd_init_size_v3(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_init_v3);
-}
-
-static u32 get_pmu_perfmon_cmd_init_size_v2(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_init_v2);
-}
-
-static u32 get_pmu_perfmon_cmd_init_size_v1(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_init_v1);
-}
-
-static u32 get_pmu_perfmon_cmd_init_size_v0(void)
-{
-        return sizeof(struct pmu_perfmon_cmd_init_v0);
-}
-
-static int get_perfmon_cmd_init_offsetofvar_v3(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_init_v3,
-                counter_alloc);
-        default:
-                return -EINVAL;
-        }
-        return 0;
-}
-
-static int get_perfmon_cmd_init_offsetofvar_v2(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_init_v2,
-                counter_alloc);
-        default:
-                return -EINVAL;
-                break;
-        }
-        return 0;
-}
-
-static int get_perfmon_cmd_init_offsetofvar_v1(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_init_v1,
-                counter_alloc);
-        default:
-                return -EINVAL;
-                break;
-        }
-        return 0;
-}
-
-static int get_perfmon_cmd_init_offsetofvar_v0(
-        enum pmu_perfmon_cmd_start_fields field)
-{
-        switch (field) {
-        case COUNTER_ALLOC:
-                return offsetof(struct pmu_perfmon_cmd_init_v0,
-                counter_alloc);
-        default:
-                return -EINVAL;
-                break;
-        }
-        return 0;
-}
-
-static void perfmon_start_set_cmd_type_v3(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
-
-        start->cmd_type = value;
-}
-
-static void perfmon_start_set_cmd_type_v2(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
-        start->cmd_type = value;
-}
-
-static void perfmon_start_set_cmd_type_v1(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
-        start->cmd_type = value;
-}
-
-static void perfmon_start_set_cmd_type_v0(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
-        start->cmd_type = value;
-}
-
-static void perfmon_start_set_group_id_v3(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
-
-        start->group_id = value;
-}
-
-static void perfmon_start_set_group_id_v2(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
-        start->group_id = value;
-}
-
-static void perfmon_start_set_group_id_v1(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
-        start->group_id = value;
-}
-
-static void perfmon_start_set_group_id_v0(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
-        start->group_id = value;
-}
-
-static void perfmon_start_set_state_id_v3(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
-
-        start->state_id = value;
-}
-
-static void perfmon_start_set_state_id_v2(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
-        start->state_id = value;
-}
-
-static void perfmon_start_set_state_id_v1(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
-        start->state_id = value;
-}
-
-static void perfmon_start_set_state_id_v0(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
-        start->state_id = value;
-}
-
-static void perfmon_start_set_flags_v3(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
-
-        start->flags = value;
-}
-
-static void perfmon_start_set_flags_v2(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
-        start->flags = value;
-}
-
-static void perfmon_start_set_flags_v1(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
-        start->flags = value;
-}
-
-static void perfmon_start_set_flags_v0(struct pmu_perfmon_cmd *pc, u8 value)
-{
-        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
-        start->flags = value;
-}
-
-static u8 perfmon_start_get_flags_v3(struct pmu_perfmon_cmd *pc)
-{
-        struct pmu_perfmon_cmd_start_v3 *start = &pc->start_v3;
-
-        return start->flags;
-}
-
-static u8 perfmon_start_get_flags_v2(struct pmu_perfmon_cmd *pc)
-{
-        struct pmu_perfmon_cmd_start_v2 *start = &pc->start_v2;
-        return start->flags;
-}
-
-static u8 perfmon_start_get_flags_v1(struct pmu_perfmon_cmd *pc)
-{
-        struct pmu_perfmon_cmd_start_v1 *start = &pc->start_v1;
-        return start->flags;
-}
-
-static u8 perfmon_start_get_flags_v0(struct pmu_perfmon_cmd *pc)
-{
-        struct pmu_perfmon_cmd_start_v0 *start = &pc->start_v0;
-        return start->flags;
-}
-
-static void perfmon_cmd_init_set_sample_buffer_v3(struct pmu_perfmon_cmd *pc,
-        u16 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->sample_buffer = value;
-}
-
-static void perfmon_cmd_init_set_sample_buffer_v2(struct pmu_perfmon_cmd *pc,
-        u16 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->sample_buffer = value;
-}
-
-
-static void perfmon_cmd_init_set_sample_buffer_v1(struct pmu_perfmon_cmd *pc,
-        u16 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->sample_buffer = value;
-}
-
-static void perfmon_cmd_init_set_sample_buffer_v0(struct pmu_perfmon_cmd *pc,
-        u16 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->sample_buffer = value;
-}
-
-static void perfmon_cmd_init_set_dec_cnt_v3(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->to_decrease_count = value;
-}
-
-static void perfmon_cmd_init_set_dec_cnt_v2(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->to_decrease_count = value;
-}
-
-static void perfmon_cmd_init_set_dec_cnt_v1(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->to_decrease_count = value;
-}
-
-static void perfmon_cmd_init_set_dec_cnt_v0(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->to_decrease_count = value;
-}
-
-static void perfmon_cmd_init_set_base_cnt_id_v3(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->base_counter_id = value;
-}
-
-static void perfmon_cmd_init_set_base_cnt_id_v2(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->base_counter_id = value;
-}
-
-static void perfmon_cmd_init_set_base_cnt_id_v1(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->base_counter_id = value;
-}
-
-static void perfmon_cmd_init_set_base_cnt_id_v0(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->base_counter_id = value;
-}
-
-static void perfmon_cmd_init_set_samp_period_us_v3(struct pmu_perfmon_cmd *pc,
-        u32 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->sample_period_us = value;
-}
-
-static void perfmon_cmd_init_set_samp_period_us_v2(struct pmu_perfmon_cmd *pc,
-        u32 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->sample_period_us = value;
-}
-
-static void perfmon_cmd_init_set_samp_period_us_v1(struct pmu_perfmon_cmd *pc,
-        u32 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->sample_period_us = value;
-}
-
-static void perfmon_cmd_init_set_samp_period_us_v0(struct pmu_perfmon_cmd *pc,
-        u32 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->sample_period_us = value;
-}
-
-static void perfmon_cmd_init_set_num_cnt_v3(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->num_counters = value;
-}
-
-static void perfmon_cmd_init_set_num_cnt_v2(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->num_counters = value;
-}
-
-static void perfmon_cmd_init_set_num_cnt_v1(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->num_counters = value;
-}
-
-static void perfmon_cmd_init_set_num_cnt_v0(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->num_counters = value;
-}
-
-static void perfmon_cmd_init_set_mov_avg_v3(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v3 *init = &pc->init_v3;
-
-        init->samples_in_moving_avg = value;
-}
-
-static void perfmon_cmd_init_set_mov_avg_v2(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v2 *init = &pc->init_v2;
-        init->samples_in_moving_avg = value;
-}
-
-static void perfmon_cmd_init_set_mov_avg_v1(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v1 *init = &pc->init_v1;
-        init->samples_in_moving_avg = value;
-}
-
-static void perfmon_cmd_init_set_mov_avg_v0(struct pmu_perfmon_cmd *pc,
-        u8 value)
-{
-        struct pmu_perfmon_cmd_init_v0 *init = &pc->init_v0;
-        init->samples_in_moving_avg = value;
-}
-
-static void get_pmu_init_msg_pmu_queue_params_v0(struct pmu_queue *queue,
-        u32 id, void *pmu_init_msg)
-{
-        struct pmu_init_msg_pmu_v0 *init =
-                (struct pmu_init_msg_pmu_v0 *)pmu_init_msg;
-        queue->index    = init->queue_info[id].index;
-        queue->offset   = init->queue_info[id].offset;
-        queue->size = init->queue_info[id].size;
-}
-
-static void get_pmu_init_msg_pmu_queue_params_v1(struct pmu_queue *queue,
-        u32 id, void *pmu_init_msg)
-{
-        struct pmu_init_msg_pmu_v1 *init =
-                (struct pmu_init_msg_pmu_v1 *)pmu_init_msg;
-        queue->index    = init->queue_info[id].index;
-        queue->offset   = init->queue_info[id].offset;
-        queue->size = init->queue_info[id].size;
-}
-
-static void get_pmu_init_msg_pmu_queue_params_v2(struct pmu_queue *queue,
-        u32 id, void *pmu_init_msg)
-{
-        struct pmu_init_msg_pmu_v2 *init =
-                (struct pmu_init_msg_pmu_v2 *)pmu_init_msg;
-        queue->index    = init->queue_info[id].index;
-        queue->offset   = init->queue_info[id].offset;
-        queue->size = init->queue_info[id].size;
-}
-
-static void get_pmu_init_msg_pmu_queue_params_v4(struct pmu_queue *queue,
-        u32 id, void *pmu_init_msg)
-{
-        struct pmu_init_msg_pmu_v4 *init = pmu_init_msg;
-        u32 current_ptr = 0;
-        u8 i;
-        u8 tmp_id = id;
-
-        if (tmp_id == PMU_COMMAND_QUEUE_HPQ)
-                tmp_id = PMU_QUEUE_HPQ_IDX_FOR_V3;
-        else if (tmp_id == PMU_COMMAND_QUEUE_LPQ)
-                tmp_id = PMU_QUEUE_LPQ_IDX_FOR_V3;
-        else if (tmp_id == PMU_MESSAGE_QUEUE)
-                tmp_id = PMU_QUEUE_MSG_IDX_FOR_V3;
-        else
-                return;
-
-        queue->index    = init->queue_index[tmp_id];
-        queue->size = init->queue_size[tmp_id];
-        if (tmp_id != 0) {
-                for (i = 0 ; i < tmp_id; i++)
-                        current_ptr += init->queue_size[i];
-        }
-        queue->offset   = init->queue_offset + current_ptr;
-}
-static void get_pmu_init_msg_pmu_queue_params_v3(struct pmu_queue *queue,
-        u32 id, void *pmu_init_msg)
-{
-        struct pmu_init_msg_pmu_v3 *init =
-                (struct pmu_init_msg_pmu_v3 *)pmu_init_msg;
-        u32 current_ptr = 0;
-        u8 i;
-        u8 tmp_id = id;
-
-        if (tmp_id == PMU_COMMAND_QUEUE_HPQ)
-                tmp_id = PMU_QUEUE_HPQ_IDX_FOR_V3;
-        else if (tmp_id == PMU_COMMAND_QUEUE_LPQ)
-                tmp_id = PMU_QUEUE_LPQ_IDX_FOR_V3;
-        else if (tmp_id == PMU_MESSAGE_QUEUE)
-                tmp_id = PMU_QUEUE_MSG_IDX_FOR_V3;
-        else
-                return;
-        queue->index    = init->queue_index[tmp_id];
-        queue->size = init->queue_size[tmp_id];
-        if (tmp_id != 0) {
-                for (i = 0 ; i < tmp_id; i++)
-                        current_ptr += init->queue_size[i];
-        }
-        queue->offset   = init->queue_offset + current_ptr;
-}
-
-static void *get_pmu_sequence_in_alloc_ptr_v3(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->in_v3);
-}
-
-static void *get_pmu_sequence_in_alloc_ptr_v1(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->in_v1);
-}
-
-static void *get_pmu_sequence_in_alloc_ptr_v0(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->in_v0);
-}
-
-static void *get_pmu_sequence_out_alloc_ptr_v3(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->out_v3);
-}
-
-static void *get_pmu_sequence_out_alloc_ptr_v1(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->out_v1);
-}
-
-static void *get_pmu_sequence_out_alloc_ptr_v0(struct pmu_sequence *seq)
-{
-        return (void *)(&seq->out_v0);
-}
-
-static u8 pg_cmd_eng_buf_load_size_v0(struct pmu_pg_cmd *pg)
-{
-        return sizeof(pg->eng_buf_load_v0);
-}
-
-static u8 pg_cmd_eng_buf_load_size_v1(struct pmu_pg_cmd *pg)
-{
-        return sizeof(pg->eng_buf_load_v1);
-}
-
-static u8 pg_cmd_eng_buf_load_size_v2(struct pmu_pg_cmd *pg)
-{
-        return sizeof(pg->eng_buf_load_v2);
-}
-
-static void pg_cmd_eng_buf_load_set_cmd_type_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.cmd_type = value;
-}
-
-static void pg_cmd_eng_buf_load_set_cmd_type_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.cmd_type = value;
-}
-
-static void pg_cmd_eng_buf_load_set_cmd_type_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.cmd_type = value;
-}
-
-static void pg_cmd_eng_buf_load_set_engine_id_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.engine_id = value;
-}
-static void pg_cmd_eng_buf_load_set_engine_id_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.engine_id = value;
-}
-static void pg_cmd_eng_buf_load_set_engine_id_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.engine_id = value;
-}
-static void pg_cmd_eng_buf_load_set_buf_idx_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.buf_idx = value;
-}
-static void pg_cmd_eng_buf_load_set_buf_idx_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.buf_idx = value;
-}
-static void pg_cmd_eng_buf_load_set_buf_idx_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.buf_idx = value;
-}
-
-static void pg_cmd_eng_buf_load_set_pad_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.pad = value;
-}
-static void pg_cmd_eng_buf_load_set_pad_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.pad = value;
-}
-static void pg_cmd_eng_buf_load_set_pad_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.pad = value;
-}
-
-static void pg_cmd_eng_buf_load_set_buf_size_v0(struct pmu_pg_cmd *pg,
-        u16 value)
-{
-        pg->eng_buf_load_v0.buf_size = value;
-}
-static void pg_cmd_eng_buf_load_set_buf_size_v1(struct pmu_pg_cmd *pg,
-        u16 value)
-{
-        pg->eng_buf_load_v1.dma_desc.dma_size = value;
-}
-static void pg_cmd_eng_buf_load_set_buf_size_v2(struct pmu_pg_cmd *pg,
-        u16 value)
-{
-        pg->eng_buf_load_v2.dma_desc.params = value;
-}
-
-static void pg_cmd_eng_buf_load_set_dma_base_v0(struct pmu_pg_cmd *pg,
-        u32 value)
-{
-        pg->eng_buf_load_v0.dma_base = (value >> 8);
-}
-static void pg_cmd_eng_buf_load_set_dma_base_v1(struct pmu_pg_cmd *pg,
-        u32 value)
-{
-        pg->eng_buf_load_v1.dma_desc.dma_addr.lo |= u64_lo32(value);
-        pg->eng_buf_load_v1.dma_desc.dma_addr.hi |= u64_hi32(value);
-}
-static void pg_cmd_eng_buf_load_set_dma_base_v2(struct pmu_pg_cmd *pg,
-        u32 value)
-{
-        pg->eng_buf_load_v2.dma_desc.address.lo = u64_lo32(value);
-        pg->eng_buf_load_v2.dma_desc.address.hi = u64_lo32(value);
-}
-
-static void pg_cmd_eng_buf_load_set_dma_offset_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.dma_offset = value;
-}
-static void pg_cmd_eng_buf_load_set_dma_offset_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.dma_desc.dma_addr.lo |= value;
-}
-static void pg_cmd_eng_buf_load_set_dma_offset_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.dma_desc.address.lo |= u64_lo32(value);
-        pg->eng_buf_load_v2.dma_desc.address.hi |= u64_lo32(value);
-}
-
-static void pg_cmd_eng_buf_load_set_dma_idx_v0(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v0.dma_idx = value;
-}
-static void pg_cmd_eng_buf_load_set_dma_idx_v1(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v1.dma_desc.dma_idx = value;
-}
-static void pg_cmd_eng_buf_load_set_dma_idx_v2(struct pmu_pg_cmd *pg,
-        u8 value)
-{
-        pg->eng_buf_load_v2.dma_desc.params |= (value << 24);
-}
-
-int gk20a_init_pmu(struct nvgpu_pmu *pmu)
-{
-        struct gk20a *g = gk20a_from_pmu(pmu);
-        struct pmu_v *pv = &g->ops.pmu_ver;
-        int err;
-
-        err = nvgpu_mutex_init(&pmu->elpg_mutex);
-        if (err)
-                return err;
-
-        err = nvgpu_mutex_init(&pmu->pg_mutex);
-        if (err)
-                goto fail_elpg;
-
-        err = nvgpu_mutex_init(&pmu->isr_mutex);
-        if (err)
-                goto fail_pg;
-
-        err = nvgpu_mutex_init(&pmu->pmu_copy_lock);
-        if (err)
-                goto fail_isr;
-
-        err = nvgpu_mutex_init(&pmu->pmu_seq_lock);
-        if (err)
-                goto fail_pmu_copy;
-
-        pmu->remove_support = gk20a_remove_pmu_support;
-
-        switch (pmu->desc->app_version) {
-        case APP_VERSION_NC_2:
-        case APP_VERSION_NC_1:
-        case APP_VERSION_NC_0:
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
-                                pg_cmd_eng_buf_load_size_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
-                                pg_cmd_eng_buf_load_set_cmd_type_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v1;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v1;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v4;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v4;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v4;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v4;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v4;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v4;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v4;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v2;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v2;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v2;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v2;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v2;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v2;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v2;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v1;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v2;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v2;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v2;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v2;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v2;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v2;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v2;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v2;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v2;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v2;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v1;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v1;
-                break;
-        case APP_VERSION_NC_3:
-                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 =
-                                pg_cmd_eng_buf_load_set_cmd_type_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = false;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v6;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v5;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v5;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v3;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v3;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v3;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v3;
-                g->ops.pmu_ver.pmu_allocation_get_fb_addr =
-                                pmu_allocation_get_fb_addr_v3;
-                g->ops.pmu_ver.pmu_allocation_get_fb_size =
-                                pmu_allocation_get_fb_size_v3;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v4;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v4;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v4;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v4;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v3;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v3;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v3;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v3;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v3;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v3;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v3;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v3;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v3;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v3;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v3;
-                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 =
-                                pg_cmd_eng_buf_load_set_cmd_type_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v2;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v5;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v5;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v5;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v3;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v3;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v3;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v3;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v3;
-                g->ops.pmu_ver.pmu_allocation_get_fb_addr =
-                                pmu_allocation_get_fb_addr_v3;
-                g->ops.pmu_ver.pmu_allocation_get_fb_size =
-                                pmu_allocation_get_fb_size_v3;
-                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 =
-                                        get_pmu_msg_pmu_init_msg_ptr_v2;
-                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                                        get_pmu_init_msg_pmu_sw_mg_off_v2;
-                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                                        get_pmu_init_msg_pmu_sw_mg_size_v2;
-                }
-                else
-                {
-                        g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                                get_pmu_init_msg_pmu_queue_params_v3;
-                        g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                                get_pmu_msg_pmu_init_msg_ptr_v3;
-                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                                get_pmu_init_msg_pmu_sw_mg_off_v3;
-                        g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                                get_pmu_init_msg_pmu_sw_mg_size_v3;
-                }
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v3;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v3;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v3;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v3;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v3;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v3;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v3;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v3;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v3;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v3;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v3;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v3;
-                break;
-        case APP_VERSION_GM20B_5:
-        case APP_VERSION_GM20B_4:
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
-                                pg_cmd_eng_buf_load_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
-                                pg_cmd_eng_buf_load_set_cmd_type_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v0;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v3;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v3;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v3;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v3;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v3;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v3;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v3;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v1;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v1;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v1;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v1;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v1;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v1;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v1;
-                break;
-        case APP_VERSION_GM20B_3:
-        case APP_VERSION_GM20B_2:
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
-                                pg_cmd_eng_buf_load_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
-                                pg_cmd_eng_buf_load_set_cmd_type_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v0;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v2;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v2;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v2;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v2;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v2;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v2;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v2;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v2;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v2;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v2;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v1;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v1;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v1;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v1;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v1;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v1;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v1;
-                break;
-        case APP_VERSION_GM20B_1:
-        case APP_VERSION_GM20B:
-        case APP_VERSION_1:
-        case APP_VERSION_2:
-        case APP_VERSION_3:
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
-                                pg_cmd_eng_buf_load_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
-                                pg_cmd_eng_buf_load_set_cmd_type_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v0;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 16;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v0;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v0;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v1;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v1;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        set_pmu_cmdline_args_secure_mode_v1;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_size =
-                        set_pmu_cmdline_args_falctracesize_v1;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_base =
-                        set_pmu_cmdline_args_falctracedmabase_v1;
-                g->ops.pmu_ver.set_pmu_cmdline_args_trace_dma_idx =
-                        set_pmu_cmdline_args_falctracedmaidx_v1;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v1;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v1;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v1;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v1;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v1;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v1;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v1;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v1;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v1;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v1;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v1;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v1;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v1;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v1;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v1;
-                break;
-        case APP_VERSION_0:
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_size =
-                                pg_cmd_eng_buf_load_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_cmd_type =
-                                pg_cmd_eng_buf_load_set_cmd_type_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_engine_id =
-                                pg_cmd_eng_buf_load_set_engine_id_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_idx =
-                                pg_cmd_eng_buf_load_set_buf_idx_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_pad =
-                                pg_cmd_eng_buf_load_set_pad_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_buf_size =
-                                pg_cmd_eng_buf_load_set_buf_size_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_base =
-                                pg_cmd_eng_buf_load_set_dma_base_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_offset =
-                                pg_cmd_eng_buf_load_set_dma_offset_v0;
-                g->ops.pmu_ver.pg_cmd_eng_buf_load_set_dma_idx =
-                                pg_cmd_eng_buf_load_set_dma_idx_v0;
-                g->ops.pmu_ver.cmd_id_zbc_table_update = 14;
-                g->ops.pmu_ver.is_pmu_zbc_save_supported = true;
-                g->ops.pmu_ver.get_perfmon_cntr_ptr = get_perfmon_cntr_ptr_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_ut = set_perfmon_cntr_ut_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_lt = set_perfmon_cntr_lt_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_valid =
-                        set_perfmon_cntr_valid_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_index =
-                        set_perfmon_cntr_index_v0;
-                g->ops.pmu_ver.set_perfmon_cntr_group_id =
-                        set_perfmon_cntr_group_id_v0;
-                g->ops.pmu_ver.get_perfmon_cntr_sz = pmu_perfmon_cntr_sz_v0;
-                g->ops.pmu_ver.get_pmu_cmdline_args_size =
-                        pmu_cmdline_size_v0;
-                g->ops.pmu_ver.set_pmu_cmdline_args_cpu_freq =
-                        set_pmu_cmdline_args_cpufreq_v0;
-                g->ops.pmu_ver.set_pmu_cmdline_args_secure_mode =
-                        NULL;
-                g->ops.pmu_ver.get_pmu_cmdline_args_ptr =
-                        get_pmu_cmdline_args_ptr_v0;
-                g->ops.pmu_ver.get_pmu_allocation_struct_size =
-                        get_pmu_allocation_size_v0;
-                g->ops.pmu_ver.set_pmu_allocation_ptr =
-                        set_pmu_allocation_ptr_v0;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_size =
-                        pmu_allocation_set_dmem_size_v0;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_size =
-                        pmu_allocation_get_dmem_size_v0;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset =
-                        pmu_allocation_get_dmem_offset_v0;
-                g->ops.pmu_ver.pmu_allocation_get_dmem_offset_addr =
-                        pmu_allocation_get_dmem_offset_addr_v0;
-                g->ops.pmu_ver.pmu_allocation_set_dmem_offset =
-                        pmu_allocation_set_dmem_offset_v0;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
-                        get_pmu_init_msg_pmu_queue_params_v0;
-                g->ops.pmu_ver.get_pmu_msg_pmu_init_msg_ptr =
-                        get_pmu_msg_pmu_init_msg_ptr_v0;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_off =
-                        get_pmu_init_msg_pmu_sw_mg_off_v0;
-                g->ops.pmu_ver.get_pmu_init_msg_pmu_sw_mg_size =
-                        get_pmu_init_msg_pmu_sw_mg_size_v0;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_start_size =
-                        get_pmu_perfmon_cmd_start_size_v0;
-                g->ops.pmu_ver.get_perfmon_cmd_start_offsetofvar =
-                        get_perfmon_cmd_start_offsetofvar_v0;
-                g->ops.pmu_ver.perfmon_start_set_cmd_type =
-                        perfmon_start_set_cmd_type_v0;
-                g->ops.pmu_ver.perfmon_start_set_group_id =
-                        perfmon_start_set_group_id_v0;
-                g->ops.pmu_ver.perfmon_start_set_state_id =
-                        perfmon_start_set_state_id_v0;
-                g->ops.pmu_ver.perfmon_start_set_flags =
-                        perfmon_start_set_flags_v0;
-                g->ops.pmu_ver.perfmon_start_get_flags =
-                        perfmon_start_get_flags_v0;
-                g->ops.pmu_ver.get_pmu_perfmon_cmd_init_size =
-                        get_pmu_perfmon_cmd_init_size_v0;
-                g->ops.pmu_ver.get_perfmon_cmd_init_offsetofvar =
-                        get_perfmon_cmd_init_offsetofvar_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_sample_buffer =
-                        perfmon_cmd_init_set_sample_buffer_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_dec_cnt =
-                        perfmon_cmd_init_set_dec_cnt_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_base_cnt_id =
-                        perfmon_cmd_init_set_base_cnt_id_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_samp_period_us =
-                        perfmon_cmd_init_set_samp_period_us_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_num_cnt =
-                        perfmon_cmd_init_set_num_cnt_v0;
-                g->ops.pmu_ver.perfmon_cmd_init_set_mov_avg =
-                        perfmon_cmd_init_set_mov_avg_v0;
-                g->ops.pmu_ver.get_pmu_seq_in_a_ptr =
-                        get_pmu_sequence_in_alloc_ptr_v0;
-                g->ops.pmu_ver.get_pmu_seq_out_a_ptr =
-                        get_pmu_sequence_out_alloc_ptr_v0;
-                break;
-        default:
-                nvgpu_err(g, "PMU code version not supported version: %d",
-                        pmu->desc->app_version);
-                err = -EINVAL;
-                goto fail_pmu_seq;
-        }
-        pv->set_perfmon_cntr_index(pmu, 3); /* GR & CE2 */
-        pv->set_perfmon_cntr_group_id(pmu, PMU_DOMAIN_GROUP_PSTATE);
-
-        return 0;
-
-fail_pmu_seq:
-        nvgpu_mutex_destroy(&pmu->pmu_seq_lock);
-fail_pmu_copy:
-        nvgpu_mutex_destroy(&pmu->pmu_copy_lock);
-fail_isr:
-        nvgpu_mutex_destroy(&pmu->isr_mutex);
-fail_pg:
-        nvgpu_mutex_destroy(&pmu->pg_mutex);
-fail_elpg:
-        nvgpu_mutex_destroy(&pmu->elpg_mutex);
-        return err;
-}
 
 void pmu_copy_from_dmem(struct nvgpu_pmu *pmu,
                 u32 src, u8 *dst, u32 size, u8 port)
@@ -2793,63 +692,6 @@ void gk20a_pmu_msgq_tail(struct nvgpu_pmu *pmu, u32 *tail, bool set)
                         pwr_pmu_msgq_tail_val_f(*tail));
 }
 
-void gk20a_remove_pmu_support(struct nvgpu_pmu *pmu)
-{
-        struct gk20a *g = gk20a_from_pmu(pmu);
-
-        gk20a_dbg_fn("");
-
-        if (nvgpu_alloc_initialized(&pmu->dmem))
-                nvgpu_alloc_destroy(&pmu->dmem);
-
-        nvgpu_release_firmware(g, pmu->fw);
-
-        nvgpu_mutex_destroy(&pmu->elpg_mutex);
-        nvgpu_mutex_destroy(&pmu->pg_mutex);
-        nvgpu_mutex_destroy(&pmu->isr_mutex);
-        nvgpu_mutex_destroy(&pmu->pmu_copy_lock);
-        nvgpu_mutex_destroy(&pmu->pmu_seq_lock);
-}
-
-static int gk20a_prepare_ucode(struct gk20a *g)
-{
-        struct nvgpu_pmu *pmu = &g->pmu;
-        int err = 0;
-        struct mm_gk20a *mm = &g->mm;
-        struct vm_gk20a *vm = mm->pmu.vm;
-
-        if (pmu->fw)
-                return gk20a_init_pmu(pmu);
-
-        pmu->fw = nvgpu_request_firmware(g, GK20A_PMU_UCODE_IMAGE, 0);
-        if (!pmu->fw) {
-                nvgpu_err(g, "failed to load pmu ucode!!");
-                return err;
-        }
-
-        gk20a_dbg_fn("firmware loaded");
-
-        pmu->desc = (struct pmu_ucode_desc *)pmu->fw->data;
-        pmu->ucode_image = (u32 *)((u8 *)pmu->desc +
-                        pmu->desc->descriptor_size);
-
-        err = nvgpu_dma_alloc_map_sys(vm, GK20A_PMU_UCODE_SIZE_MAX,
-                        &pmu->ucode);
-        if (err)
-                goto err_release_fw;
-
-        nvgpu_mem_wr_n(g, &pmu->ucode, 0, pmu->ucode_image,
-                        pmu->desc->app_start_offset + pmu->desc->app_size);
-
-        return gk20a_init_pmu(pmu);
-
- err_release_fw:
-        nvgpu_release_firmware(g, pmu->fw);
-        pmu->fw = NULL;
-
-        return err;
-}
-
 static void pmu_handle_pg_buf_config_msg(struct gk20a *g, struct pmu_msg *msg,
                         void *param, u32 handle, u32 status)
 {
@@ -3018,7 +860,7 @@ u32 gk20a_pmu_pg_feature_list(struct gk20a *g, u32 pg_engine_id)
 void gk20a_init_pmu_ops(struct gpu_ops *gops)
 {
         gops->pmu.is_pmu_supported = gk20a_is_pmu_supported;
-        gops->pmu.prepare_ucode = gk20a_prepare_ucode;
+        gops->pmu.prepare_ucode = nvgpu_pmu_prepare_ns_ucode_blob;
         gops->pmu.pmu_setup_hw_and_bootstrap = gk20a_init_pmu_setup_hw1;
         gops->pmu.pmu_nsbootstrap = pmu_bootstrap;
         gops->pmu.pmu_get_queue_head = pwr_pmu_queue_head_r;
diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
index 1d2e20e6..b5038bd4 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.h
@@ -30,24 +30,6 @@ struct nvgpu_firmware;
 
 #define ZBC_MASK(i)                     (~(~(0) << ((i)+1)) & 0xfffe)
 
-#define APP_VERSION_NC_3        22204331
-#define APP_VERSION_NC_2        20429989
-#define APP_VERSION_NC_1        20313802
-#define APP_VERSION_NC_0        20360931
-#define APP_VERSION_GM206       20652057
-#define APP_VERSION_NV_GPU      21307569
-#define APP_VERSION_NV_GPU_1    21308030
-#define APP_VERSION_GM20B_5 20490253
-#define APP_VERSION_GM20B_4 19008461
-#define APP_VERSION_GM20B_3 18935575
-#define APP_VERSION_GM20B_2 18694072
-#define APP_VERSION_GM20B_1 18547257
-#define APP_VERSION_GM20B 17615280
-#define APP_VERSION_3 18357968
-#define APP_VERSION_2 18542378
-#define APP_VERSION_1 17997577 /*Obsolete this once 18357968 gets in*/
-#define APP_VERSION_0 16856675
-
 /*Fuse defines*/
 #if LINUX_VERSION_CODE < KERNEL_VERSION(4, 4, 0)
 #define FUSE_GCPLEX_CONFIG_FUSE_0           0x2C8
@@ -127,11 +109,7 @@ void pmu_copy_from_dmem(struct nvgpu_pmu *pmu,
                 u32 src, u8 *dst, u32 size, u8 port);
 int pmu_reset(struct nvgpu_pmu *pmu);
 int pmu_bootstrap(struct nvgpu_pmu *pmu);
-int gk20a_init_pmu(struct nvgpu_pmu *pmu);
 void pmu_dump_falcon_stats(struct nvgpu_pmu *pmu);
-void gk20a_remove_pmu_support(struct nvgpu_pmu *pmu);
-
-int gk20a_init_pmu(struct nvgpu_pmu *pmu);
 
 int gk20a_pmu_ap_send_command(struct gk20a *g,
                 union pmu_ap_cmd *p_ap_cmd, bool b_block);
diff --git a/drivers/gpu/nvgpu/gm20b/acr_gm20b.c b/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
index 580ba5e5..02da3de9 100644
--- a/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
+++ b/drivers/gpu/nvgpu/gm20b/acr_gm20b.c
@@ -161,7 +161,7 @@ static int pmu_ucode_details(struct gk20a *g, struct flcn_ucode_img *p_img)
         pmu->ucode_image = (u32 *)pmu_fw->data;
         g->acr.pmu_desc = pmu_desc;
 
-        err = gk20a_init_pmu(pmu);
+        err = nvgpu_init_pmu_fw_support(pmu);
         if (err) {
                 gm20b_dbg_pmu("failed to set function pointers\n");
                 goto release_sig;
@@ -395,7 +395,7 @@ int prepare_ucode_blob(struct gk20a *g)
         if (g->acr.ucode_blob.cpu_va) {
                 /*Recovery case, we do not need to form
                 non WPR blob of ucodes*/
-                err = gk20a_init_pmu(pmu);
+                err = nvgpu_init_pmu_fw_support(pmu);
                 if (err) {
                         gm20b_dbg_pmu("failed to set function pointers\n");
                         return err;
diff --git a/drivers/gpu/nvgpu/gp106/acr_gp106.c b/drivers/gpu/nvgpu/gp106/acr_gp106.c
index 46bc5055..076c8829 100644
--- a/drivers/gpu/nvgpu/gp106/acr_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/acr_gp106.c
@@ -174,7 +174,7 @@ static int pmu_ucode_details(struct gk20a *g, struct flcn_ucode_img_v1 *p_img)
         pmu->ucode_image = (u32 *)pmu_fw->data;
         g->acr.pmu_desc = pmu_desc;
 
-        err = gk20a_init_pmu(pmu);
+        err = nvgpu_init_pmu_fw_support(pmu);
         if (err) {
                 nvgpu_err(g, "failed to set function pointers");
                 goto release_sig;
@@ -389,7 +389,7 @@ static int gp106_prepare_ucode_blob(struct gk20a *g)
         if (g->acr.ucode_blob.cpu_va) {
                 /*Recovery case, we do not need to form
                 non WPR blob of ucodes*/
-                err = gk20a_init_pmu(pmu);
+                err = nvgpu_init_pmu_fw_support(pmu);
                 if (err) {
                         gp106_dbg_pmu("failed to set function pointers\n");
                         return err;
diff --git a/drivers/gpu/nvgpu/include/nvgpu/pmu.h b/drivers/gpu/nvgpu/include/nvgpu/pmu.h
index 169d8b98..15f37bda 100644
--- a/drivers/gpu/nvgpu/include/nvgpu/pmu.h
+++ b/drivers/gpu/nvgpu/include/nvgpu/pmu.h
@@ -401,4 +401,8 @@ int nvgpu_pmu_vidmem_surface_alloc(struct gk20a *g, struct nvgpu_mem *mem,
 int nvgpu_pmu_sysmem_surface_alloc(struct gk20a *g, struct nvgpu_mem *mem,
                 u32 size);
 
+/* PMU F/W support */
+int nvgpu_init_pmu_fw_support(struct nvgpu_pmu *pmu);
+int nvgpu_pmu_prepare_ns_ucode_blob(struct gk20a *g);
+
 #endif /* __NVGPU_PMU_H__ */