Yuan Luo 968e181826 Fix triton_fused_moe unit test and benchmark (#9276) ...

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>

2025-08-18 00:54:33 -07:00

..

ascend

…

configs

…

cpu

…

entrypoints/http_server

…

ep

…

hicache

…

lora

…

models

…

openai_server

…

quant

…

rl

…

double-sparsity-config-Llama-3.1-8B-Instruct.json

…

experiment_runner.py

…

kv_cache_scales_llama3_1_8b.json

…

kv_cache_scales_llama3_8b.json

…

kv_cache_scales_qwen2_1_5b.json

…

parse_results.py

…

run_suite.py

…

test_abort.py

…

test_ascend_graph_tp1_bf16.py

…

test_ascend_graph_tp2_bf16.py

…

test_ascend_w8a8_quantization.py

…

test_bench_one_batch.py

…

test_bench_serving.py

…

test_bnb.py

…

test_chunked_prefill.py

…

test_cpp_radix_cache.py

…

test_create_kvindices.py

…

test_custom_allreduce.py

…

test_data_parallelism.py

…

test_disaggregation_different_tp.py

…

test_disaggregation_pp.py

…

test_disaggregation.py

…

test_double_sparsity.py

…

test_dp_attention.py

…

test_eagle_infer_a.py

…

test_eagle_infer_b.py

…

test_ebnf_constrained.py

…

test_eval_accuracy_large.py

…

test_eval_fp8_accuracy.py

…

test_expert_distribution.py

…

test_expert_location_updater.py

…

test_fa3.py

…

test_fim_completion.py

…

test_flashmla.py

…

test_forward_split_prefill.py

…

test_full_deepseek_v3.py

…

test_function_call_parser.py

…

test_fused_moe.py

…

test_get_weights_by_name.py

…

test_gguf.py

…

test_gpt_oss_1gpu.py

…

test_gpt_oss_4gpu.py

…

test_gpt_oss_common.py

…

test_gptqmodel_dynamic.py

…

test_health_check.py

…

test_hidden_states.py

…

test_hybrid_attn_backend.py

…

test_input_embeddings.py

…

test_intel_amx_attention_backend.py

…

test_io_struct.py

…

test_jinja_template_utils.py

…

test_kv_events.py

…

test_local_attn.py

…

test_metrics.py

…

test_mla_deepseek_v3.py

…

test_mla_flashinfer.py

…

test_mla_fp8.py

…

test_mla_int8_deepseek_v3.py

…

test_mla_tp.py

…

test_mla.py

…

test_modelopt_fp8kvcache.py

…

test_modelopt.py

…

test_models_from_modelscope.py

…

test_moe_eval_accuracy_large.py

…

test_mscclpp.py

…

test_multi_instance_release_memory_occupation.py

…

test_nightly_gsm8k_eval_amd.py

…

test_nightly_gsm8k_eval.py

…

test_no_chunked_prefill.py

…

test_no_overlap_scheduler.py

…

test_page_size.py

…

test_patch_torch.py

…

test_penalty.py

…

test_pp_single_node.py

…

test_pytorch_sampling_backend.py

…

test_quick_allreduce.py

…

test_radix_attention.py

…

test_reasoning_parser.py

…

test_regex_constrained.py

…

test_release_memory_occupation.py

…

test_request_queue_validation.py

…

test_retract_decode.py

…

test_rope_rocm.py

…

test_sagemaker_server.py

…

test_schedule_policy.py

…

test_score_api.py

…

test_server_args.py

…

test_session_control.py

…

test_skip_tokenizer_init.py

…

test_srt_endpoint.py

…

test_srt_engine_with_quant_args.py

…

test_srt_engine.py

…

test_start_profile.py

…

test_swa_unittest.py

…

test_tokenizer_batch_encode.py

…

test_torch_compile_moe.py

…

test_torch_compile.py

…

test_torch_native_attention_backend.py

…

test_torch_tp.py

…

test_torchao.py

…

test_triton_attention_backend.py

…

test_triton_attention_kernels.py

…

test_triton_attention_rocm_mla.py

…

test_triton_fused_moe.py

Fix triton_fused_moe unit test and benchmark (#9276)

2025-08-18 00:54:33 -07:00

test_triton_moe_channel_fp8_kernel.py

…

test_triton_moe_wna16.py

…

test_triton_sliding_window.py

…

test_two_batch_overlap.py

…

test_utils_update_weights.py

…

test_vertex_endpoint.py

…

test_vision_chunked_prefill.py

…

test_vision_openai_server_a.py

…

test_vision_openai_server_b.py

…

test_vision_openai_server_common.py

…

test_vllm_dependency.py

…

test_vlm_accuracy.py

…

test_vlm_input_format.py

…

test_wave_attention_backend.py

…

test_wave_attention_kernels.py

…

test_weight_version.py

…