class_docstr: 使用 OneflowSingleDriver 测试基类的函数
class_docstr_tok: ["\u4f7f\u7528","OneflowSingleDriver","\u6d4b\u8bd5\u57fa\u7c7b\u7684\u51fd\u6570"]
docstr: 测试move_model_to_device函数
docstr_tok: ["\u6d4b\u8bd5move_model_to_device\u51fd\u6570"]
code: def test_move_model_to_device_cpu(self):
      """
      测试move_model_to_device函数
      """
      OneflowSingleDriver.move_model_to_device(self.driver.model, "cpu")
      assert self.driver.model.linear1.weight.device.type == "cpu"
  
imports: import pytest
  from pathlib import Path
  from fastNLP.core.drivers.oneflow_driver.single_device import OneflowSingleDriver
  from fastNLP.core.samplers import ReproduceBatchSampler, RandomSampler
  from tests.helpers.models.oneflow_model import OneflowNormalModel_Classification_1
  from tests.helpers.datasets.oneflow_data import OneflowNormalDataset, OneflowNormalXYDataset
  from tests.helpers.datasets.torch_data import TorchNormalDataset
  from tests.helpers.models.torch_model import TorchNormalModel_Classification_1
  from fastNLP.envs.distributed import rank_zero_rm
  from fastNLP.envs.imports import _NEED_IMPORT_ONEFLOW, _NEED_IMPORT_TORCH
  from fastNLP import prepare_oneflow_dataloader, BucketedBatchSampler
label_desc: Use image node_id 9 for calling the TestOneflowDriverFunctions obj's underlying member method code with example usage: obj.test_move_model_to_device_cpu() without return types