frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
main
pytorch/docs/source/mps.md
Abhinav Tharamel d41f62b7a0 Fix/issue #155027 (#155252) 
Fixes #155027
Converted RST files to Markdown

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155252
Approved by: https://github.com/svekars

Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
2025-06-08 21:17:31 +00:00

993 B
Raw Permalink Blame History

torch.mps

.. automodule:: torch.mps

.. currentmodule:: torch.mps

.. autosummary::
    :toctree: generated
    :nosignatures:

    device_count
    synchronize
    get_rng_state
    set_rng_state
    manual_seed
    seed
    empty_cache
    set_per_process_memory_fraction
    current_allocated_memory
    driver_allocated_memory
    recommended_max_memory
    compile_shader

MPS Profiler

.. autosummary::
    :toctree: generated
    :nosignatures:

    profiler.start
    profiler.stop
    profiler.profile

    profiler.is_capturing_metal
    profiler.is_metal_capture_enabled
    profiler.metal_capture

MPS Event

.. autosummary::
    :toctree: generated
    :nosignatures:

    event.Event

% This module needs to be documented. Adding here in the meantime

% for tracking purposes

.. py:module:: torch.mps.event

.. py:module:: torch.mps.profiler