code: def _test_setup_module_move_to_device(fabric, move_to_device):
      model = torch.nn.Linear(10, 10, bias=False)
      with mock.patch(
          "lightning.fabric.wrappers._FabricModule"
      ) as fabric_module_mock:
          fabric_model = fabric.setup_module(
              model, move_to_device=move_to_device
          )
      fabric_module_mock.assert_not_called()
  
      # The _DeviceDtypeModuleMixin currently can't represent the device in a meaningful way for models with pieces on
      # different devices
      assert fabric_model.device.type == "xla"
      assert fabric.device.type == "xla"
  
file_path: pytorch-lightning/tests/tests_fabric/strategies/test_xla_fsdp_integration.py
imports: import os
  import re
  from pathlib import Path
  from unittest import mock
  import pytest
  import torch
  from lightning.fabric import Fabric
  from lightning.fabric.strategies import XLAFSDPStrategy
  from torch.utils.data import DataLoader
  from tests_fabric.helpers.models import RandomDataset
  from tests_fabric.helpers.runif import RunIf
label_desc: Use image node_id 5 for calling a global function with example usage: _test_setup_module_move_to_device(fabric, move_to_device) without return types