code: def test_basic_cpu():
      """Test single batch behavior on CPU"""
      model = Linear(2, 2, bias=False)
      try:
          optim = AdaScale(SGD(model.parameters(), lr=0.1))
      except RuntimeError:
          return
      assert False, "Single batch AdaScale should not be suppported"
  
imports: import tempfile
  import numpy
  import pytest
  import torch
  from torch import Tensor
  from torch.nn import Linear, Sequential
  from torch.optim import SGD
  from torch.optim.lr_scheduler import LambdaLR
  from fairscale.fair_dev.testing.golden_testing_data import adascale_test_data
  from fairscale.fair_dev.testing.testing import make_cudnn_deterministic, skip_if_no_cuda
  from fairscale.fair_dev.testing.testing_memory import find_tensor_by_shape
  from fairscale.optim import AdaScale
label_desc: Use image node_id 1 for calling a global function with example usage: test_basic_cpu() without return types