update accelerator registry

justusschock · justusschock · commit 3e58f9bd71b2 · 2022-07-14T10:40:18.000+02:00
diff --git a/tests/tests_pytorch/accelerators/test_accelerator_registry.py b/tests/tests_pytorch/accelerators/test_accelerator_registry.py
@@ -63,4 +63,4 @@ def is_available():
 
 
 def test_available_accelerators_in_registry():
-    assert AcceleratorRegistry.available_accelerators() == ["cpu", "gpu", "hpu", "ipu", "mps", "tpu"]
+    assert AcceleratorRegistry.available_accelerators() == ["cpu", "cuda", "tpu", "hpu", "ipu", "mps", "tpu"]

Original file line number	Diff line number	Diff line change
`@@ -63,4 +63,4 @@ def is_available():`
`63`	`63`
`64`	`64`
`65`	`65`	`def test_available_accelerators_in_registry():`
`66`		`- assert AcceleratorRegistry.available_accelerators() == ["cpu", "gpu", "hpu", "ipu", "mps", "tpu"]`
	`66`	`+ assert AcceleratorRegistry.available_accelerators() == ["cpu", "cuda", "tpu", "hpu", "ipu", "mps", "tpu"]`