code: def get_zero_files(dir):
      file_list = get_files(dir)
      for prefix in [
          ZERO_FILE_PREFIX,
          FP16_ZERO_FILE_PREFIX,
          BF16_ZERO_FILE_PREFIX,
      ]:
          zero_files = get_files_with_prefix(file_list, prefix)
          if len(zero_files) > 0:
              return zero_files
  
      return []
  
file_path: deepspeed/deepspeed/checkpoint/reshape_utils.py
imports: import os
  import torch
  from collections import OrderedDict
  from .constants import ZERO_FILE_PREFIX, FP16_ZERO_FILE_PREFIX, BF16_ZERO_FILE_PREFIX
label_desc: Use image node_id 5 for calling a global function with example usage: get_zero_files(dir) and returns: list, zero_files