We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce9e326 commit e735382Copy full SHA for e735382
backends/aoti/aoti_backend.py
@@ -39,13 +39,13 @@ class AotiBackend(BackendDetails, ABC):
39
Base backend class for AOTInductor-based backends.
40
41
This class provides common functionality for compiling models using AOTInductor
42
- with different device targets (CUDA, Metal/MPS, etc.).
+ with different device targets (CUDA, Metal, etc.).
43
"""
44
45
@staticmethod
46
@abstractmethod
47
def get_device_name() -> str:
48
- """Return the device name for this backend (e.g., 'cuda', 'mps')."""
+ """Return the device name for this backend (e.g., 'cuda', 'metal')."""
49
pass
50
51
0 commit comments