1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/pmu/pmu_fw.c b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
index 8c4e091d..20eadb50 100644
--- a/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
+++ b/drivers/gpu/nvgpu/common/pmu/pmu_fw.c
@@ -2226,6 +2226,8 @@ static int nvgpu_init_pmu_fw_ver_ops(struct nvgpu_pmu *pmu)
 static void nvgpu_remove_pmu_support(struct nvgpu_pmu *pmu)
 {
         struct gk20a *g = gk20a_from_pmu(pmu);
+        struct mm_gk20a *mm = &g->mm;
+        struct vm_gk20a *vm = mm->pmu.vm;
 
         nvgpu_log_fn(g, " ");
 
@@ -2247,6 +2249,11 @@ static void nvgpu_remove_pmu_support(struct nvgpu_pmu *pmu)
         if (g->acr.hsbl_fw)
                 nvgpu_release_firmware(g, g->acr.hsbl_fw);
 
+        nvgpu_dma_unmap_free(vm, &g->acr.acr_ucode);
+        nvgpu_dma_unmap_free(vm, &g->acr.hsbl_ucode);
+
+        nvgpu_dma_unmap_free(vm, &pmu->seq_buf);
+
         nvgpu_mutex_destroy(&pmu->elpg_mutex);
         nvgpu_mutex_destroy(&pmu->pg_mutex);
         nvgpu_mutex_destroy(&pmu->isr_mutex);