c869a153创建于 2025年7月24日历史提交
文件最后提交记录最后更新时间
!82 整改CMake工程,支持bisheng编译器;解决安全编译问题 Merge pull request !82 from yuantao/cmake 10 个月前
!177 更新Docs指南 Merge pull request !177 from 徐天昊/docs 10 个月前
!103 examples9~10适配为device调用 * examples9~10适配为device调用 1 年前
README.md

GroupedMatmulSliceKPerTokenDequant Example Readme

代码组织

├── 11_grouped_matmul_slice_k_per_token_dequant
│   ├── CMakeLists.txt     # CMake编译文件
│   ├── README.md
│   └── 11_grouped_matmul_slice_k_per_token_dequant.cpp # 主文件

功能介绍

该算子支持A矩阵在k轴切分,和B矩阵按照group分组进行矩阵乘。之后进行per token的量化操作。 A/B矩阵为int8类型,scale为bf16,输出结果为bf16

使用示例

  • 获取代码之后编译相应的算子可执行文件,可参考quickstart
  • 执行算子
# 编译指定用例
bash scripts/build.sh 11_grouped_matmul_slice_k_per_token_dequant
# cd [代码仓路径]/output/bin
# 可执行文件名|group数量|矩阵m轴|n轴|k轴|Device ID
# Device ID可选,默认为0
./11_grouped_matmul_slice_k_per_token_dequant 128 512 1024 2048 0

执行结果如下,说明精度比对成功。

Compare success.