torch.mps.recommended_max_memory

torch.mps.recommended_max_memory() [source]

Returns recommended max Working set size for GPU memory in bytes.

Note

Recommended max working set size for Metal. returned from device.recommendedMaxWorkingSetSize.