code: def get_alignment_size(x: Tensor) -> int:
      if (
          x.dtype == torch.float16
          or x.dtype == torch.half
          or x.dtype == torch.bfloat16
      ):
          return 8
      elif x.dtype == torch.float32 or x.dtype == torch.float:
          return 4
      else:
          return 0
  
imports: import functools
  from typing import List, Optional
  import torch
  from torch import Tensor
  from torch._inductor import utils
  from torch.utils._mode_utils import no_dispatch
  from torch.utils._triton import has_triton
  from ..pattern_matcher import fwd_only, joint_fwd_bwd, Match, register_replacement
label_desc: Use image node_id 3 for calling a global function with example usage: get_alignment_size(x) and returns: int, int, int