Skip to content

Commit 5b11bdb

Browse files
committed
mps override decorator.
1 parent 0a5725b commit 5b11bdb

File tree

1 file changed

+1
-0
lines changed
  • src/lightning/pytorch/accelerators

1 file changed

+1
-0
lines changed

src/lightning/pytorch/accelerators/mps.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
8888
)
8989

9090
@classmethod
91+
@override
9192
def device_name(cls, device: Optional[_DEVICE] = None) -> str:
9293
# todo: implement a better way to get the device name
9394
available = cls.is_available()

0 commit comments

Comments
 (0)