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

[doc] Improve mps package description (#104184)

Browse Source 
Fixes #104183

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104184
Approved by: https://github.com/malfet
This commit is contained in:
David Radley
2023-06-27 15:50:36 +00:00
committed by PyTorch MergeBot
parent ad2905ad27
commit 4d613b9a5f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/mps/__init__.py
View File

@ -1,5 +1,8 @@
r"""
This package enables an interface for accessing MPS backend in python
This package enables an interface for accessing MPS (Metal Performance Shaders) backend in Python.
Metal is Apple's API for programming metal GPU (graphics processor unit). Using MPS means that increased
performance can be achieved, by running work on the metal GPU(s).
See https://developer.apple.com/documentation/metalperformanceshaders for more details.
"""
import torch
from .. import Tensor