torch.mps.recommended_max_memory

torch.mps.recommended_max_memory()

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

Note

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

Return type

int