Init the main branch for aiter

85762c1a · Xiaowei.zhang · ae0b3521 · 85762c1a · 85762c1a · 85762c1a
Commit 85762c1a authored Apr 15, 2026 by Xiaowei.zhang
20 changed files
--- a/csrc/include/torch/mha_v3_varlen_bwd.h
+++ b/csrc/include/torch/mha_v3_varlen_bwd.h
--- a/csrc/include/torch/mha_varlen_bwd.h
+++ b/csrc/include/torch/mha_varlen_bwd.h
--- a/csrc/include/torch/mha_varlen_fwd.h
+++ b/csrc/include/torch/mha_varlen_fwd.h
--- a/csrc/include/vectorization.cuh
+++ b/csrc/include/vectorization.cuh
--- a/csrc/include/warp_sort.h
+++ b/csrc/include/warp_sort.h
--- a/csrc/kernels/activation_kernels.cu
+++ b/csrc/kernels/activation_kernels.cu
--- a/csrc/kernels/attention_ragged.cu
+++ b/csrc/kernels/attention_ragged.cu
--- a/csrc/kernels/attention_v1.cu
+++ b/csrc/kernels/attention_v1.cu
--- a/csrc/kernels/binary_operator.cu
+++ b/csrc/kernels/binary_operator.cu
--- a/csrc/kernels/cache_kernels.cu
+++ b/csrc/kernels/cache_kernels.cu
--- a/csrc/kernels/custom_all_reduce.cu
+++ b/csrc/kernels/custom_all_reduce.cu
--- a/csrc/kernels/custom_kernels.cu
+++ b/csrc/kernels/custom_kernels.cu
--- a/csrc/kernels/fused_kernels.cu
+++ b/csrc/kernels/fused_kernels.cu
--- a/csrc/kernels/fused_qk_norm.cu
+++ b/csrc/kernels/fused_qk_norm.cu
--- a/csrc/kernels/fused_qk_norm_mrope_cache_quant.cu
+++ b/csrc/kernels/fused_qk_norm_mrope_cache_quant.cu
--- a/csrc/kernels/fused_qk_norm_rope_cache_quant.cu
+++ b/csrc/kernels/fused_qk_norm_rope_cache_quant.cu
--- a/csrc/kernels/generate_binaryop.py
+++ b/csrc/kernels/generate_binaryop.py
--- a/csrc/kernels/mha_common.cu
+++ b/csrc/kernels/mha_common.cu
--- a/csrc/kernels/moe_align_sum_kernels.cu
+++ b/csrc/kernels/moe_align_sum_kernels.cu
--- a/csrc/kernels/moe_fused_gate.cu
+++ b/csrc/kernels/moe_fused_gate.cu