zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add a mx.metal.device_info (#1060)

Browse Source 
* device inof

* add variant

* fix linux

* fix doc
This commit is contained in:
Awni Hannun
2024-04-30 15:47:27 -07:00
committed by GitHub
parent a30e7ed2da
commit 19bef39f5c
5 changed files with 41 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/src/python/metal.rst
View File

@@ -7,6 +7,7 @@ Metal
:toctree: _autosummary
is_available
device_info
get_active_memory
get_peak_memory
get_cache_memory