Skip to content

Commit 62153b3

Browse files
awaelchlilantiga
authored andcommitted
Delete test that was removed on master
1 parent ceb44fc commit 62153b3

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tests/tests_fabric/strategies/test_deepspeed_integration.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -276,22 +276,6 @@ def test_deepspeed_env_variables_on_platforms(_, deepspeed_dist_mock, platform):
276276
assert os.environ["LOCAL_RANK"] == str(strategy.local_rank)
277277

278278

279-
@RunIf(min_cuda_gpus=2, standalone=True, deepspeed=True)
280-
def test_deepspeed_specific_gpu_device_index():
281-
"""Test that the DeepSpeed strategy can run on specific device indices."""
282-
283-
class RunFabric(BoringFabric):
284-
def step(self, model, batch):
285-
assert self.device.type == "cuda"
286-
assert self.device.index == 1
287-
assert batch.device.index == 1
288-
assert model.device.index == 1
289-
return super().step(model, batch)
290-
291-
fabric = RunFabric(accelerator="cuda", devices=[1], strategy="deepspeed")
292-
fabric.run()
293-
294-
295279
@RunIf(min_cuda_gpus=2, standalone=True, deepspeed=True, bf16_cuda=True)
296280
def test_deepspeed_with_bfloat16_precision():
297281
"""Test that the DeepSpeed strategy works with bfloat16 precision."""

0 commit comments

Comments
 (0)