BasicConv2dTla Example Readme
代码组织
├── 56_ascend950_basic_conv2d_tla
│ ├── CMakeLists.txt # CMake编译文件
│ ├── README.md
│ └── basic_conv2d_tla.cpp # 主文件
功能说明
- 算子功能:完成基础卷积计算
示例说明
该用例总体设计与_basic_matmul相同,区别为使用了TLA相关抽象,因此提供相关示例说明
使用示例
- 获取代码之后编译相应的算子可执行文件,可参考quickstart,本用例为Ascend 950算子,编译时需加-DCATLASS_ARCH=3510
- 执行算子
# 编译指定用例
bash scripts/build.sh 56_ascend950_basic_conv2d_tla -DCATLASS_ARCH=3510
cd ./output/bin
# 可执行文件名 |Batch|Hi|Wi|Cin|Cout|kh|kw|padL|padR|padT|padB|strideH|strideW|dilationH|dilationW|Device ID
# Device ID可选,默认为0
./56_ascend950_basic_conv2d_tla 2 33 43 112 80 3 3 2 2 2 2 1 1 1 1 0
执行结果如下,表明精度验证通过。
Compare success.