[RFC 7/8] drm/nouveau: set max supported vGPU count when SRIOV is supported

[Danilo Krummrich]

On Fri, Nov 22, 2024 at 04:57:11AM -0800, Zhi Wang wrote:
> Set the max supported vGPU count according to the number of VFs when
> SRIOV is supported on Ada.
> 
> Suggested-by: Jason Gunthorpe <jgg at nvidia.com>
> Cc: Surath Mitra <smitra at nvidia.com>
> Signed-off-by: Zhi Wang <zhiw at nvidia.com>
> ---
>  drivers/gpu/drm/nouveau/include/nvkm/subdev/gsp.h | 1 +
>  drivers/gpu/drm/nouveau/nvkm/subdev/gsp/ad102.c   | 4 +++-
>  drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c    | 1 +
>  3 files changed, 5 insertions(+), 1 deletion(-)
> 
> diff --git a/drivers/gpu/drm/nouveau/include/nvkm/subdev/gsp.h b/drivers/gpu/drm/nouveau/include/nvkm/subdev/gsp.h
> index c6fe2d9d47de..6e244af1e815 100644
> --- a/drivers/gpu/drm/nouveau/include/nvkm/subdev/gsp.h
> +++ b/drivers/gpu/drm/nouveau/include/nvkm/subdev/gsp.h
> @@ -64,6 +64,7 @@ struct nvkm_gsp {
>  			} frts, boot, elf, heap;
>  			u64 addr;
>  			u64 size;
> +			u64 max_vgpu_count;
>  		} wpr2;
>  		struct {
>  			u64 addr;
> diff --git a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/ad102.c b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/ad102.c
> index 1e403dbd7323..80d6d73fe352 100644
> --- a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/ad102.c
> +++ b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/ad102.c
> @@ -69,8 +69,10 @@ ad102_gsp_init_fw_heap(struct nvkm_gsp *gsp)
>  	num_vfs = pci_sriov_get_totalvfs(device_pci->pdev);
>  	if (!num_vfs)
>  		nvkm_gsp_init_fw_heap(gsp, 0);
> -	else
> +	else {
>  		nvkm_gsp_init_fw_heap(gsp, 576 * SZ_1M);
> +		gsp->fb.wpr2.max_vgpu_count = num_vfs;
> +	}

Please also add braces on the if block.

>  
>  	if (gsp->fb.wpr2.heap.size <= SZ_256M)
>  		return 0;
> diff --git a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
> index 5a47201bf0c4..2647a83773d2 100644
> --- a/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
> +++ b/drivers/gpu/drm/nouveau/nvkm/subdev/gsp/r535.c
> @@ -1968,6 +1968,7 @@ r535_gsp_wpr_meta_init(struct nvkm_gsp *gsp)
>  	meta->partitionRpcAddr = 0;
>  	meta->partitionRpcRequestOffset = 0;
>  	meta->partitionRpcReplyOffset = 0;
> +	meta->gspFwHeapVfPartitionCount = gsp->fb.wpr2.max_vgpu_count;
>  	meta->verified = 0;
>  	return 0;
>  }
> -- 
> 2.34.1
>