code: def _assert_secondary_tensor_size(model: Module) -> None:
      for _, param in model.named_parameters():
          assert param.ds_secondary_tensor is not None
          assert (
              param.ds_secondary_tensor.size()[0]
              % param.ds_tensor.size()[0]
              == 0
          )
  
file_path: DeepSpeed/tests/unit/runtime/zero/test_zeropp.py
imports: import pytest
  import deepspeed.comm
  from torch.nn import Module
  from unit.common import DistributedTest
  from unit.simple_model import random_dataloader
  import deepspeed
  from deepspeed.runtime.zero.config import DeepSpeedZeroConfig
  import torch.nn
label_desc: Use image node_id 3 for calling a global function with example usage: _assert_secondary_tensor_size(model) without return types