Skip to content

Commit 3fe2428

Browse files
committed
alias gpu to CUDAAccelerator
1 parent 2caa1af commit 3fe2428

File tree

1 file changed

+6
-0
lines changed
  • src/pytorch_lightning/accelerators

1 file changed

+6
-0
lines changed

src/pytorch_lightning/accelerators/cuda.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,12 @@ def register_accelerators(cls, accelerator_registry: Dict) -> None:
9797
cls,
9898
description=f"{cls.__class__.__name__}",
9999
)
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+
)
100106

101107
def teardown(self) -> None:
102108
# clean up memory

0 commit comments

Comments
 (0)