mirror of
https://github.com/p-e-w/heretic.git
synced 2026-07-09 17:28:29 +00:00
MPS support (#5)
* MPS support * oops, added issue tracker. * Delete .beads/issues.jsonl
This commit is contained in:
parent
b3545e4b1e
commit
09730bad70
2 changed files with 4 additions and 0 deletions
|
|
@ -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.[/]"
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue