[New Feature] Support W4Afp8 MoE GroupGemm (#3171)

* init

* 增加多线程编译

* fix bug

* fix bug

* code style

* 增加fp16

* 将print替换成assert

* 修复stmatrix

* 减小单测shape

* 减小单测shape
This commit is contained in:
yangjianfengo1
2025-08-06 10:34:05 +08:00
committed by GitHub
parent 841e831575
commit 89397516a8
10 changed files with 1455 additions and 2 deletions

3
.gitignore vendored
View File

@@ -167,3 +167,6 @@ build
.ccls-cache
third_party
custom_ops/gpu_ops/w4afp8_gemm/w4afp8_gemm_*.cu
custom_ops/gpu_ops/w4afp8_gemm/w4afp8_gemm_template.h