Files
llm_trainer/scripts/stat_utils/cal_flops.py