diff --git a/drivers/gpu/nvgpu/gp106/acr_gp106.c b/drivers/gpu/nvgpu/gp106/acr_gp106.c
index 0b4b86f45..0e49214ed 100644
--- a/drivers/gpu/nvgpu/gp106/acr_gp106.c
+++ b/drivers/gpu/nvgpu/gp106/acr_gp106.c
@@ -91,31 +91,12 @@ static void flcn64_set_dma(struct falc_u64 *dma_addr, u64 value)
 int gp106_alloc_blob_space(struct gk20a *g,
 		size_t size, struct mem_desc *mem)
 {
-	int err = 0;
 	struct wpr_carveout_info wpr_inf;
 
 	g->ops.pmu.get_wpr(g, &wpr_inf);
 
-	mem->aperture = APERTURE_VIDMEM;
-	mem->sgt = kzalloc(sizeof(*mem->sgt), GFP_KERNEL);
-	if (!mem->sgt) {
-		gk20a_err(dev_from_gk20a(g), "failed to allocate memory\n");
-		return -ENOMEM;
-	}
-
-	err = sg_alloc_table(mem->sgt, 1, GFP_KERNEL);
-	if (err) {
-		gk20a_err(dev_from_gk20a(g), "failed to allocate sg_table\n");
-		goto free_sgt;
-	}
-
-	sg_dma_address(mem->sgt->sgl) = wpr_inf.nonwpr_base;
-
-	return err;
-
-free_sgt:
-	gk20a_free_sgtable(&mem->sgt);
-	return err;
+	return gk20a_gmmu_alloc_attr_vid_at(g, 0, wpr_inf.size, mem,
+			wpr_inf.nonwpr_base);
 }
 
 void gp106_init_secure_pmu(struct gpu_ops *gops)