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

../
moe/                                               10-Sep-2025 08:16                   -
__init__.py                                        10-Sep-2025 08:16                   0
test_attention.py                                  10-Sep-2025 08:16                8192
test_attention_no_rope.py                          10-Sep-2025 08:16                8840
test_attention_packed_seq.py                       10-Sep-2025 08:16                7355
test_core_attention.py                             10-Sep-2025 08:16                2042
test_cuda_graphs.py                                10-Sep-2025 08:16                2971
test_mlp.py                                        10-Sep-2025 08:16                2253
test_module.py                                     10-Sep-2025 08:16                3745
test_multi_latent_attention.py                     10-Sep-2025 08:16               19380
test_multi_token_prediction.py                     10-Sep-2025 08:16               15405
test_relative_attention.py                         10-Sep-2025 08:16                1390
test_retro_attention.py                            10-Sep-2025 08:16                7848
test_rope.py                                       10-Sep-2025 08:16                3294
test_spec_customization.py                         10-Sep-2025 08:16               11782
test_transformer_block.py                          10-Sep-2025 08:16               17809
test_transformer_block_custom_pgs.py               10-Sep-2025 08:16               30147
test_transformer_layer.py                          10-Sep-2025 08:16                7852
test_utils.py                                      10-Sep-2025 08:16                4022