code: def _element_size(dtype):
      """
      Returns the element size for a dtype, in bytes
      """
      if not isinstance(dtype, torch.dtype):
          raise RuntimeError(
              f"expected torch.dtype, but got {type(dtype)}"
          )
  
      if dtype.is_complex:
          return torch.finfo(dtype).bits >> 2
      elif dtype.is_floating_point:
          return torch.finfo(dtype).bits >> 3
      elif dtype == torch.bool:
          # NOTE: torch.bool is not supported in torch.iinfo()
          return 1
      else:
          return torch.iinfo(dtype).bits >> 3
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/benchmarks/communication/utils.py
imports: import torch
  import os
  import math
  import argparse
  from benchmarks.communication.constants import *
label_desc: Use image node_id 11 for calling a global function with example usage: _element_size(dtype) and returns: unknown, unknown, int, unknown