code: def allclose(x, y):
      assert x.dtype == y.dtype
      rtol, atol = {
          torch.float32: (5e-4, 5e-5),
          torch.float16: (3e-2, 2e-3),
      }[x.dtype]
      return torch.allclose(x, y, rtol=rtol, atol=atol)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/ops/transformer/inference/test_bias_gelu.py
imports: import pytest
  import torch
  import deepspeed
  from deepspeed.ops.op_builder import InferenceBuilder
label_desc: Use image node_id 1 for calling a global function with example usage: allclose(x, y) and returns: torch