def test_num_stepping_batches_with_tpu_multi():
    """Test stepping batches with the TPU strategy across multiple devices."""
    trainer = Trainer(accelerator="tpu", devices="auto", max_epochs=1, logger=False, enable_checkpointing=False)
    model = MultiprocessModel()
    trainer.fit(model)
Test stepping batches with the TPU strategy across multiple devices.
tests/tests_pytorch/trainer/properties/test_estimated_stepping_batches.py