Skip to content

Commit

Permalink
And here as well
Browse files Browse the repository at this point in the history
  • Loading branch information
malfet committed Feb 17, 2023
1 parent a5e9081 commit 751e9a5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_C/__init__.pyi.in
Original file line number Diff line number Diff line change
Expand Up @@ -1207,7 +1207,7 @@ def _mps_setMemoryFraction(fraction: _float) -> None: ...
def _mps_currentAllocatedMemory() -> _int: ...
def _mps_driverAllocatedMemory() -> _int: ...
def _mps_is_available() -> _bool: ...
def _mps_is_on_macos_13_or_newer() -> _bool: ...
def _mps_is_on_macos_13_or_newer(minor: _int) -> _bool: ...

# Defined in torch/csrc/cuda/Module.cpp
def _cuda_getCurrentStream(device: _int) -> Tuple: ...
Expand Down

0 comments on commit 751e9a5

Please sign in to comment.