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 2caa1af commit 3fe2428Copy full SHA for 3fe2428
src/pytorch_lightning/accelerators/cuda.py
@@ -97,6 +97,12 @@ def register_accelerators(cls, accelerator_registry: Dict) -> None:
97
cls,
98
description=f"{cls.__class__.__name__}",
99
)
100
+ # temporarily enable "gpu" to point to the CUDA Accelerator
101
+ accelerator_registry.register(
102
+ "gpu",
103
+ cls,
104
+ description=f"{cls.__class__.__name__}",
105
+ )
106
107
def teardown(self) -> None:
108
# clean up memory
0 commit comments