diff --git a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
index 2540ab128..f041c9c13 100644
--- a/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/pmu_gk20a.c
@@ -3628,7 +3628,7 @@ static int pmu_process_init_msg(struct pmu_gk20a *pmu,
 			   pv->get_pmu_init_msg_pmu_sw_mg_size(init)) &
 			~(PMU_DMEM_ALLOC_ALIGNMENT - 1);
 		u32 size = end - start;
-		gk20a_buddy_allocator_init(&pmu->dmem, "gk20a_pmu_dmem",
+		gk20a_bitmap_allocator_init(&pmu->dmem, "gk20a_pmu_dmem",
 					   start, size,
 					   PMU_DMEM_ALLOC_ALIGNMENT, 0);
 	}