Skip to content

Commit e735382

Browse files
authored
Update metal device name
1 parent ce9e326 commit e735382

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/aoti/aoti_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,13 @@ class AotiBackend(BackendDetails, ABC):
3939
Base backend class for AOTInductor-based backends.
4040
4141
This class provides common functionality for compiling models using AOTInductor
42-
with different device targets (CUDA, Metal/MPS, etc.).
42+
with different device targets (CUDA, Metal, etc.).
4343
"""
4444

4545
@staticmethod
4646
@abstractmethod
4747
def get_device_name() -> str:
48-
"""Return the device name for this backend (e.g., 'cuda', 'mps')."""
48+
"""Return the device name for this backend (e.g., 'cuda', 'metal')."""
4949
pass
5050

5151
@staticmethod

0 commit comments

Comments
 (0)