Re: [PATCH v3 07/31] gpu: nova-core: vgpu: reserve the 48-VM WPR2 heap

[Timur Tabi <ttabi@nvidia.com>]

On Mon, 2026-09-28 at 13:28 +0300, Zhi Wang wrote:
> 
>      /// Returns the WPR heap size to reserve when vGPU is enabled.
> -    pub(crate) fn vgpu_wpr_heap_size() -> u64 {
> -        u64::from(bindings::GSP_FW_HEAP_SIZE_VGPU_DEFAULT)
> +    pub(crate) fn vgpu_wpr_heap_size(total_vfs: u16) -> u64 {
> +        u64::from(match total_vfs {
> +            2..=32 => bindings::GSP_FW_HEAP_SIZE_VGPU_DEFAULT,
> +            _ => r000_00::GSP_FW_HEAP_SIZE_VGPU_48VMS,
> +        })

What about if total_vfs is 1?  Shouldn't you also have:

	1 => GSP_FW_HEAP_SIZE_VGPU_1VM,