torch.mps

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

`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 для процесса в байтах.

MPS Profiler

`profiler.start`	Запуск отслеживания OS Signpost с бэкэнда MPS.
`profiler.stop`	Остановка генерации отслеживания OS Signpost с бэкэнда MPS.
`profiler.profile`	Контекстный менеджер для включения генерации отслеживания OS Signpost с бэкэнда MPS.

MPS Event

event.Event

Обёртка вокруг события MPS.

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