torch.mps

Создано: Feb 10, 2023 | Последнее обновление: Jul 07, 2026

Этот пакет предоставляет интерфейс для доступа к бэкенду MPS (Metal Performance Shaders) из Python. Metal — это API Apple для программирования графического процессора (GPU) Metal. Использование MPS позволяет повысить производительность за счёт выполнения задач на графических процессорах Metal. Подробнее см. на странице https://developer.apple.com/documentation/metalperformanceshaders.

`is_available`
`device_count`	Возвращает количество доступных устройств MPS.
`synchronize`	Ожидает завершения всех ядер во всех потоках на устройстве MPS.
`get_rng_state`	Возвращает состояние генератора случайных чисел в виде ByteTensor.
`set_rng_state`	Устанавливает состояние генератора случайных чисел.
`manual_seed`	Устанавливает начальное значение для генерации случайных чисел.
`seed`	Устанавливает случайное начальное значение для генерации случайных чисел.
`empty_cache`	Освобождает всю неиспользуемую кэшированную память, удерживаемую аллокатором кэша, чтобы её могли использовать другие приложения GPU.
`set_per_process_memory_fraction`	Задаёт долю памяти, ограничивающую выделение памяти процессом на устройстве MPS.
`current_allocated_memory`	Возвращает текущий объём памяти GPU, занимаемой тензорами, в байтах.
`driver_allocated_memory`	Возвращает общий объём памяти GPU, выделенной драйвером Metal для процесса, в байтах.
`recommended_max_memory`	Возвращает рекомендуемый максимальный размер рабочего набора памяти GPU в байтах.
`compile_shader`	Компилирует вычислительный шейдер из исходного кода и позволяет вызывать определённые в нём ядра из среды выполнения Python. Пример.
`load_metallib`	Загружает предварительно скомпилированную библиотеку Metal (.metallib) и возвращает объект библиотеки шейдеров, позволяющий вызывать определённые в ней ядра.

Профилировщик MPS

`profiler.start`	Запускает трассировку OS Signpost из бэкенда MPS.
`profiler.stop`	Останавливает создание трассировки OS Signpost из бэкенда MPS.
`profiler.profile`	Контекстный менеджер для включения трассировки OS Signpost из бэкенда MPS.
`profiler.is_capturing_metal`	Проверяет, выполняется ли захват Metal.
`profiler.is_metal_capture_enabled`	Проверяет, можно ли использовать контекстный менеджер `metal_capture`. Чтобы включить захват Metal, задайте переменную среды MTL_CAPTURE_ENABLED.
`profiler.metal_capture`	Контекстный менеджер, включающий захват вызовов Metal в gputrace.

Событие MPS

event.Event

Обёртка для события MPS.

© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/mps.html