Path	Lines of Code
benchmarks/big_model_inference/big_model_inference.py	105
benchmarks/big_model_inference/measures_util.py	57
benchmarks/fp8/ms_amp/ddp.py	81
benchmarks/fp8/ms_amp/distrib_deepspeed.py	118
benchmarks/fp8/ms_amp/fp8_utils.py	67
benchmarks/fp8/ms_amp/non_distributed.py	77
benchmarks/fp8/torchao/ddp.py	107
benchmarks/fp8/torchao/distrib_deepspeed.py	163
benchmarks/fp8/torchao/fp8_utils.py	66
benchmarks/fp8/torchao/fsdp.py	121
benchmarks/fp8/torchao/non_distributed.py	100
benchmarks/fp8/transformer_engine/ddp.py	95
benchmarks/fp8/transformer_engine/distrib_deepspeed.py	141
benchmarks/fp8/transformer_engine/fp8_utils.py	66
benchmarks/fp8/transformer_engine/fsdp.py	110
benchmarks/fp8/transformer_engine/non_distributed.py	87
benchmarks/fsdp2/main.py	81
benchmarks/fsdp2/measure_utils.py	82
benchmarks/fsdp2/utils.py	169
benchmarks/fsdp2/visualize.py	80
benchmarks/torch.compile/regional_compilation.py	51