torch.mps

{eval-rst}

.. automodule:: torch.mps

{eval-rst}

.. currentmodule:: torch.mps

{eval-rst}

.. 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

{eval-rst}

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

    profiler.start
    profiler.stop
    profiler.profile

    profiler.is_capturing_metal
    profiler.is_metal_capture_enabled
    profiler.metal_capture

MPS Event

{eval-rst}

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

    event.Event

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

% for tracking purposes

{eval-rst}

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

{eval-rst}

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