code: def debug_log_split_param_diff(idx, param, param_name, partitions):
      # Log some useful comparison of tensors that are being mapped.
      # Note that the global param index for layers and modules may be different but the shapes
      # and semantics of the layer should match.
      logging.debug(
          f"Index: {idx} Model Params : {param_name} - {param.shape}"
      )
      logging.debug(
          f"Index: {idx} Global params: {partitions[1][idx]} - {partitions[0][idx].shape}"
      )
  
imports: import os
  import shutil
  import tarfile
  import tempfile
  from argparse import ArgumentParser
  from typing import Dict, List
  import torch
  import torch.nn
  from omegaconf import OmegaConf, open_dict
  from pytorch_lightning import Trainer
  from nemo.collections.nlp.parts.nlp_overrides import NEMO_MEGATRON_MODEL_PARALLEL_APPSTATE_OVERRIDE, GradScaler, MegatronHalfPrecisionPlugin, NLPDDPStrategy, NLPSaveRestoreConnector, PipelineMixedPrecisionPlugin
  from nemo.utils import logging, model_utils
  from nemo.utils.app_state import AppState
label_desc: Use image node_id 7 for calling a global function with example usage: debug_log_split_param_diff(idx, param, param_name, partitions) without return types