added the utility function to print model flops#47

Open

ultrons wants to merge 3 commits intopytorch-tpu:llama2-google-next-trainingfrom ultrons:print-tflops

+32

Commits on Nov 7, 2023

added the utility function to print model flops
ultrons
committed

Commits on Nov 8, 2023

refining step time calc
ultrons
committed

Commits on Nov 9, 2023

clean up
ultrons
committed