| torch.mps | |
| =================================== | |
| .. automodule:: torch.mps | |
| .. currentmodule:: torch.mps | |
| .. autosummary:: | |
| :toctree: generated | |
| :nosignatures: | |
| synchronize | |
| get_rng_state | |
| set_rng_state | |
| manual_seed | |
| seed | |
| empty_cache | |
| set_per_process_memory_fraction | |
| current_allocated_memory | |
| driver_allocated_memory | |
| MPS Profiler | |
| ------------ | |
| .. autosummary:: | |
| :toctree: generated | |
| :nosignatures: | |
| profiler.start | |
| profiler.stop | |
| profiler.profile |