Index of /FlagScale_to_hygon/third_party/Megatron-LM/megatron/legacy/mpu/tests/

../
__init__.py                                        10-Sep-2025 08:16                   0
commons.py                                         10-Sep-2025 08:16                2058
test_cross_entropy.py                              10-Sep-2025 08:16                3390
test_data.py                                       10-Sep-2025 08:16                2430
test_initialize.py                                 10-Sep-2025 08:16                3122
test_layers.py                                     10-Sep-2025 08:16               19259
test_random.py                                     10-Sep-2025 08:16                6377