Index of /FlagScale_to_hygon/third_party/Megatron-LM/tests/unit_tests/transformer/moe/

../
__init__.py                                        10-Sep-2025 08:16                   0
conftest.py                                        10-Sep-2025 08:16                1292
test_a2a_token_dispatcher.py                       10-Sep-2025 08:16                4054
test_aux_loss.py                                   10-Sep-2025 08:16                5720
test_grouped_mlp.py                                10-Sep-2025 08:16               18057
test_moe_layer.py                                  10-Sep-2025 08:16                7971
test_moe_layer_discrepancy.py                      10-Sep-2025 08:16               10943
test_multihot_indices_converter.py                 10-Sep-2025 08:16                3575
test_routers.py                                    10-Sep-2025 08:16               11754
test_sequential_mlp.py                             10-Sep-2025 08:16                8941
test_shared_experts.py                             10-Sep-2025 08:16                4766
test_token_dispatcher.py                           10-Sep-2025 08:16               15028
test_upcycling.py                                  10-Sep-2025 08:16                6667