code: def assert_close_loose(
      a: Tensor, b: Tensor, rtol: float = 1e-3, atol: float = 1e-3
  ):
      assert_close(
          a,
          b,
          rtol=rtol,
          atol=atol,
          msg=f"Tensor not close, shape: {a.shape} vs {b.shape}, \
                                                     dtype: {a.dtype} vs {b.dtype}",
      )
  
file_path: ColossalAI/colossalai/testing/comparison.py
imports: from typing import Any, List, OrderedDict
  import torch
  import torch.distributed
  from torch import Tensor
  from torch.distributed import ProcessGroup
  from torch.testing import assert_close
  from torch.utils._pytree import tree_flatten
label_desc: Use image node_id 3 for calling a global function with example usage: assert_close_loose(a, b, rtol, atol) without return types