文件最后提交记录最后更新时间
!195 纯Cube算子编译选项调整为-cube;修复msdebug模式出现链接重定义 Merge pull request !195 from yuantao/cmake 9 个月前
!185 22_padding_splitk_matmul Merge pull request !185 from 李晟泽/padding_splitk_matmul 10 个月前
!178 新增21_basic_matmul_preload_zN Merge pull request !178 from 李晟泽/21_basic_matmul_preload_zN 10 个月前
README.md

BasicMatmulPreloadZN Example Readme

代码组织

├── 21_basic_matmul_preload_zN
│   ├── CMakeLists.txt   # CMake编译文件
│   ├── README.md
│   └── basic_matmul_preload_zN.cpp # 主文件

功能介绍

该算子在00_basic_matmul基础上支持B矩阵NZ输入(非转置使用zN,转置使用nZ),并使用MmadAtlasA2Preload的DispatchPolicy。

使用示例

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

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

Compare success.