[RFC 03/29] nvkm/vgpu: reserve a larger GSP heap when NVIDIA vGPU is enabled

[Zhi Wang]

A larger GSP heap is required when enabling NVIDIA vGPU.

Cc: Neo Jia <c...@nvidia.com>
Cc: Surath Mitra <smi...@nvidia.com>
Signed-off-by: Zhi Wang <z...@nvidia.com>
---
 drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c 
b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
index 18bae367b194..a38a6abcac6f 100644
--- a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
+++ b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
@@ -2638,7 +2638,10 @@ r535_gsp_oneinit(struct nvkm_gsp *gsp)
        gsp->fb.wpr2.elf.size = gsp->fw.len;
        gsp->fb.wpr2.elf.addr = ALIGN_DOWN(gsp->fb.wpr2.boot.addr - 
gsp->fb.wpr2.elf.size, 0x10000);
 
-       {
+       if (nvkm_vgpu_mgr_is_supported(device)) {
+               gsp->fb.wpr2.heap.size = SZ_256M;
+       } else {
+
                u32 fb_size_gb = DIV_ROUND_UP_ULL(gsp->fb.size, 1 << 30);
 
                gsp->fb.wpr2.heap.size =
-- 
2.34.1