|
..
|
|
__init__.py
|
|
gpt_neo_test_3d_deparallelization.py
|
|
gpt_neo_test_3d_inference.py
|
|
gpt_neo_test_3d_inference_with_fusion.py
|
|
gpt_neo_test_3d_training.py
|
|
gpt_neo_test_3d_training_with_ddp.py
|
|
gpt_neo_test_3d_training_with_fusion.py
|
|
gpt_neo_test_3d_training_with_zero.py
|
|
gpt_neo_test_kernel_fusion.py
|
|
gpt_neo_test_pp_deparallelization.py
|
|
gpt_neo_test_pp_inference.py
|
|
gpt_neo_test_pp_inference_with_fusion.py
|
|
gpt_neo_test_pp_training.py
|
|
gpt_neo_test_pp_training_with_ddp.py
|
|
gpt_neo_test_pp_training_with_fusion.py
|
|
gpt_neo_test_pp_training_with_zero.py
|
|
gpt_neo_test_tp_deparallelization.py
|
|
gpt_neo_test_tp_inference.py
|
|
gpt_neo_test_tp_inference_with_fusion.py
|
|
gpt_neo_test_tp_training.py
|
|
gpt_neo_test_tp_training_with_ddp.py
|
|
gpt_neo_test_tp_training_with_fusion.py
|
|
gpt_neo_test_tp_training_with_zero.py
|