import aicc-master-dev

518a5f4d · hly · c2a1b310 · 518a5f4d · 518a5f4d · 518a5f4d
Commit 518a5f4d authored Jun 09, 2026 by hly
20 changed files
--- a/csrc/flash_attn_hg/include/mla/mla_prefix_prefill.h
+++ b/csrc/flash_attn_hg/include/mla/mla_prefix_prefill.h
--- a/csrc/flash_attn_hg/include/mla/mla_pv_gemm_prefetch_k_tile16x32.h
+++ b/csrc/flash_attn_hg/include/mla/mla_pv_gemm_prefetch_k_tile16x32.h
--- a/csrc/flash_attn_hg/include/mla/mla_qk_gemm_prefetch_v_tile16x32.h
+++ b/csrc/flash_attn_hg/include/mla/mla_qk_gemm_prefetch_v_tile16x32.h
--- a/csrc/flash_attn_hg/include/mla/mla_qk_gemm_utils_tile16x32.h
+++ b/csrc/flash_attn_hg/include/mla/mla_qk_gemm_utils_tile16x32.h
--- a/csrc/flash_attn_hg/include/mla/mla_softmax.h
+++ b/csrc/flash_attn_hg/include/mla/mla_softmax.h
--- a/csrc/flash_attn_hg/include/numeric_types.h
+++ b/csrc/flash_attn_hg/include/numeric_types.h
--- a/csrc/flash_attn_hg/src/flash_bwd_launch_template.h
+++ b/csrc/flash_attn_hg/src/flash_bwd_launch_template.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b16_fa.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b16_fa.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b16_mla.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b16_mla.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b16_pa.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b16_pa.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b8_fa.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b8_fa.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b8_mla.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b8_mla.h
--- a/csrc/flash_attn_hg/src/flash_fwd_b8_pa.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_b8_pa.h
--- a/csrc/flash_attn_hg/src/flash_fwd_launch_template.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_launch_template.h
--- a/csrc/flash_attn_hg/src/flash_fwd_launch_template_pa.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_launch_template_pa.h
--- a/csrc/flash_attn_hg/src/flash_fwd_reduce.h
+++ b/csrc/flash_attn_hg/src/flash_fwd_reduce.h
--- a/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_bf16.cpp
+++ b/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_bf16.cpp
--- a/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_fp16.cpp
+++ b/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_fp16.cpp
--- a/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_prefix_prefill_bf16.cpp
+++ b/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_prefix_prefill_bf16.cpp
--- a/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_prefix_prefill_fp16.cpp
+++ b/csrc/flash_attn_hg/src/target/flash_fp8_fwd_hdim128_prefix_prefill_fp16.cpp