Skip to content

Commit

Permalink
update accelerator registry
Browse files Browse the repository at this point in the history
  • Loading branch information
justusschock committed Jul 14, 2022
1 parent 66e1f5a commit 68ef8c7
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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"]

0 comments on commit 68ef8c7

Please sign in to comment.