code: def get_tflops(model_numel, batch_size, seq_len, step_time):
      # Tflops_per_GPU = global_batch * global_numel * seq_len * 8 / #gpu
      return (
          model_numel
          * batch_size
          * seq_len
          * 8
          / 1e12
          / (step_time + 1e-12)
          / 8
      )
  
file_path: ColossalAI/examples/language/gpt/experiments/auto_parallel/auto_parallel_with_gpt.py
imports: from functools import partial
  from time import time
  import psutil
  import torch
  import transformers
  from gpt_modules import GPT2LMHeadModel, GPTLMLoss
  from colossalai.auto_parallel.tensor_shard.initialize import autoparallelize
  from colossalai.initialize import launch_from_torch
  from colossalai.legacy.core import global_context
  from colossalai.logging import disable_existing_loggers, get_dist_logger
label_desc: Use image node_id 4 for calling a global function with example usage: get_tflops(model_numel, batch_size, seq_len, step_time) and returns: unknown