MPS support (#5)

* 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
+2
View File
@@ -88,6 +88,8 @@ def run():
print(f"MUSA type: [bold]{torch.musa.get_device_name()}[/]")
elif is_npu_available():
print(f"CANN version: [bold]{torch.version.cann}[/]")
elif torch.backends.mps.is_available():
print(f"GPU type: [bold]Apple Metal (MPS)[/]")
else:
print(
"[bold yellow]No GPU or other accelerator detected. Operations will be slow.[/]"
+2
View File
@@ -58,6 +58,8 @@ def empty_cache():
torch.sdaa.empty_cache()
elif is_musa_available():
torch.musa.empty_cache()
elif torch.backends.mps.is_available():
torch.mps.empty_cache()
gc.collect()