sglang_v0.5.2/flashinfer_0.3.1/tests
hailin 06e45b5ff9 local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
..
alibi_reference.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
conftest.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
jit_utils.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
rope_reference.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
sink_attention_reference.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_activation.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_alibi.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_attention_sink.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_attention_sink_blackwell.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_batch_attention.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_batch_decode_kernels.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_batch_prefill.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_batch_prefill_kernels.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_blackwell_fmha.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_block_sparse.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_block_sparse_indices_to_vector_sparse_offsets.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_bmm_fp8.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_create_ipc_buffer.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_cudnn_decode.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_cudnn_prefill.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_cudnn_prefill_deepseek.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_cute_dsl_blockscaled_gemm.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_decode_fp8_calibration_scale.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_decode_prefill_lse.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_deepseek_mla.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_fp4_quantize.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_fp4_tensor_torch_cute.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_fp8_prefill.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_fp8_quantize.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_green_ctx.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_group_gemm.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_groupwise_scaled_gemm_fp8.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_groupwise_scaled_gemm_mxfp4.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_hopper.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_hopper_fp8_attention.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_jit_example.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_jit_warmup.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_logits_cap.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_logits_processor.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_mla_decode_kernel.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_mla_page.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_mm_fp4.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_mnnvl_custom_comm.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_mnnvl_memory.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_non_contiguous_decode.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_non_contiguous_prefill.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_norm.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_nvshmem.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_nvshmem_allreduce.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_page.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_pod_kernels.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_quantization.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_rope.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_sampling.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_shared_prefix_kernels.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_single_prefill.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_sliding_window.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_sm_constraint_gemm.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_tensor_cores_decode.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_triton_cascade.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_allreduce.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_allreduce_fusion.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_alltoall.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_cutlass_fused_moe.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_gen_attention.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_gen_fused_moe.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_gen_mla.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_mnnvl_allreduce.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_moe_allreduce_fusion.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_trtllm_moe_allreduce_fusion_finalize.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_vllm_custom_allreduce.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
test_xqa.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00
utils_fp4.py local source code for flashinfer_0.3.1 && torch vision_0.22.1 2025-09-20 12:33:58 +08:00