mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
0be53d83fc
- Added torch.mps.profiler.[start() and stop()] APIs with RST documentation - Added test case in test_mps Pull Request resolved: https://github.com/pytorch/pytorch/pull/101002 Approved by: https://github.com/malfet
28 lines
495 B
ReStructuredText
28 lines
495 B
ReStructuredText
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
|