1 files changed, 12 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/pmu/pmu_fw.c b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
index 6a038317..1070579e 100644
--- a/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
+++ b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
@@ -1299,6 +1299,12 @@ static int nvgpu_init_pmu_fw_ver_ops(struct nvgpu_pmu *pmu)
                                 boardobjgrp_pmugetstatus_impl_v1;
                         g->ops.pmu_ver.boardobj.is_boardobjgrp_pmucmd_id_valid =
                                 is_boardobjgrp_pmucmd_id_valid_v1;
+                        g->ops.pmu_ver.volt.volt_set_voltage =
+                                nvgpu_volt_set_voltage_gv10x;
+                        g->ops.pmu_ver.volt.volt_get_voltage =
+                                nvgpu_volt_rail_get_voltage_gv10x;
+                        g->ops.pmu_ver.volt.volt_send_load_cmd_to_pmu =
+                                nvgpu_volt_send_load_cmd_to_pmu_gv10x;
                 } else {
                         g->ops.pmu_ver.get_pmu_init_msg_pmu_queue_params =
                                 get_pmu_init_msg_pmu_queue_params_v4;
@@ -1458,6 +1464,12 @@ static int nvgpu_init_pmu_fw_ver_ops(struct nvgpu_pmu *pmu)
                         boardobjgrp_pmugetstatus_impl;
                 g->ops.pmu_ver.boardobj.is_boardobjgrp_pmucmd_id_valid =
                         is_boardobjgrp_pmucmd_id_valid_v0;
+                g->ops.pmu_ver.volt.volt_set_voltage =
+                        nvgpu_volt_set_voltage_gp10x;
+                g->ops.pmu_ver.volt.volt_get_voltage =
+                        nvgpu_volt_rail_get_voltage_gp10x;
+                g->ops.pmu_ver.volt.volt_send_load_cmd_to_pmu =
+                        nvgpu_volt_send_load_cmd_to_pmu_gp10x;
                 break;
         case APP_VERSION_GM20B:
                 g->ops.pmu_ver.pg_cmd_eng_buf_load_size =