文件最后提交记录最后更新时间
!195 纯Cube算子编译选项调整为-cube;修复msdebug模式出现链接重定义 Merge pull request !195 from yuantao/cmake 9 个月前
!177 更新Docs指南 Merge pull request !177 from 徐天昊/docs 10 个月前
!100 0516 Migrate Device_layer feature to 00_basic_matmul/01_batched_matmul/02_grouped_matmul_slice_m * Merge branch 'master' of gitee.com:ascend/catlass into device_layer012 * cleancode * fix macro definition * migrate 00 01 02 1 年前
README.md

BatchedMatmul Example Readme

代码组织

├── 01_batched_matmul
│   ├── CMakeLists.txt     # CMake编译文件
│   ├── README.md
│   └── batched_matmul.cpp # 主文件

使用示例

  • 获取代码之后编译相应的算子可执行文件,可参考quickstart
  • 执行算子
# 编译指定用例
bash scripts/build.sh 01_batched_matmul
# cd [代码仓路径]/output/bin
# 可执行文件名 batch轴|m轴|n轴|k轴|Device ID
# Device ID可选,默认为0
# 注意!这里相比basicMatmul多一个batch轴的输入参数
./01_batched_matmul 5 256 512 1024 0

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

Compare success.