tests lite

justusschock · justusschock · commit d5d1b7a4135c · 2022-07-19T14:48:09.000+02:00
diff --git a/tests/tests_pytorch/lite/test_lite.py b/tests/tests_pytorch/lite/test_lite.py
@@ -313,9 +313,11 @@ def test_setup_dataloaders_replace_standard_sampler(shuffle, strategy):
     "accelerator, expected",
     [
         ("cpu", "cpu"),
+        pytest.param("cuda", "cuda:0", marks=RunIf(min_cuda_gpus=1)),
         pytest.param("gpu", "cuda:0", marks=RunIf(min_cuda_gpus=1)),
         pytest.param("tpu", "xla:0", marks=RunIf(tpu=True)),
         pytest.param("mps", "mps:0", marks=RunIf(mps=True)),
+        pytest.param("gpu", "mps:0", marks=RunIf(mps=True)),
     ],
 )
 def test_to_device(accelerator, expected):

Original file line number	Diff line number	Diff line change
`@@ -313,9 +313,11 @@ def test_setup_dataloaders_replace_standard_sampler(shuffle, strategy):`
`313`	`313`	`"accelerator, expected",`
`314`	`314`	`[`
`315`	`315`	`("cpu", "cpu"),`
	`316`	`+ pytest.param("cuda", "cuda:0", marks=RunIf(min_cuda_gpus=1)),`
`316`	`317`	`pytest.param("gpu", "cuda:0", marks=RunIf(min_cuda_gpus=1)),`
`317`	`318`	`pytest.param("tpu", "xla:0", marks=RunIf(tpu=True)),`
`318`	`319`	`pytest.param("mps", "mps:0", marks=RunIf(mps=True)),`
	`320`	`+ pytest.param("gpu", "mps:0", marks=RunIf(mps=True)),`
`319`	`321`	`],`
`320`	`322`	`)`
`321`	`323`	`def test_to_device(accelerator, expected):`