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.