Files
ColossalAI/examples/language/bert/benchmark_utils.py