Ty_Tech/heretic
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

MPS support (#5)

Browse Source 
* MPS support

* oops, added issue tracker.

* Delete .beads/issues.jsonl
This commit is contained in:
Matt Barnson
2025-11-17 05:12:01 -08:00
committed by GitHub
parent b3545e4b1e
commit 09730bad70
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/heretic/main.py
+2
View File
@@ -88,6 +88,8 @@ def run():
print(f"MUSA type: [bold]{torch.musa.get_device_name()}[/]") print(f"MUSA type: [bold]{torch.musa.get_device_name()}[/]")
elif is_npu_available(): elif is_npu_available():
print(f"CANN version: [bold]{torch.version.cann}[/]") print(f"CANN version: [bold]{torch.version.cann}[/]")
elif torch.backends.mps.is_available():
print(f"GPU type: [bold]Apple Metal (MPS)[/]")
else: else:
print( print(
"[bold yellow]No GPU or other accelerator detected. Operations will be slow.[/]" "[bold yellow]No GPU or other accelerator detected. Operations will be slow.[/]"
src/heretic/utils.py
+2
View File
@@ -58,6 +58,8 @@ def empty_cache():
torch.sdaa.empty_cache() torch.sdaa.empty_cache()
elif is_musa_available(): elif is_musa_available():
torch.musa.empty_cache() torch.musa.empty_cache()
elif torch.backends.mps.is_available():
torch.mps.empty_cache()
gc.collect() gc.collect()