1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
index 1409d279..c89e910c 100644
--- a/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
+++ b/drivers/gpu/nvgpu/gp10b/pmu_gp10b.c
@@ -413,6 +413,10 @@ void gp10b_init_pmu_ops(struct gpu_ops *gops)
         gops->pmu.pmu_get_queue_head_size = pwr_pmu_queue_head__size_1_v;
         gops->pmu.pmu_get_queue_tail = pwr_pmu_queue_tail_r;
         gops->pmu.pmu_get_queue_tail_size = pwr_pmu_queue_tail__size_1_v;
+        gops->pmu.pmu_queue_head = gk20a_pmu_queue_head;
+        gops->pmu.pmu_queue_tail = gk20a_pmu_queue_tail;
+        gops->pmu.pmu_mutex_acquire = gk20a_pmu_mutex_acquire;
+        gops->pmu.pmu_mutex_release = gk20a_pmu_mutex_release;
         gops->pmu.lspmuwprinitdone = false;
         gops->pmu.fecsbootstrapdone = false;
         gops->pmu.write_dmatrfbase = gp10b_write_dmatrfbase;