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 0a5725b commit 5b11bdbCopy full SHA for 5b11bdb
src/lightning/pytorch/accelerators/mps.py
@@ -88,6 +88,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
88
)
89
90
@classmethod
91
+ @override
92
def device_name(cls, device: Optional[_DEVICE] = None) -> str:
93
# todo: implement a better way to get the device name
94
available = cls.is_available()
0 commit comments