GetTilingFunc
功能说明
根据算子类型OpType获取对应的Tiling函数。
函数原型
1 | TilingFunc GetTilingFunc(const char *opType) const |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
opType |
输入 |
算子类型,与原型定义时的算子类型保持一致。 |
返回值说明
获取成功,则返回对应的Tiling函数指针;失败则返回空指针。Tiling函数指针定义如下:
1 | using TilingFunc = uint32_t(*)(gert::TilingContext*) |
约束说明
无
调用示例
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | context_ascendc::OpTilingRegistry tmpIns; bool flag = tmpIns.LoadTilingLibrary("/your/path/to/so_path/liboptiling.so"); if (flag == false) { std::cout << "Load tiling so failed" << std::endl; return -1; } context_ascendc::TilingFunc tilingFunc = tmpIns.GetTilingFunc("AddCustom"); if (tilingFunc != nullptr) { // use tiling func ... } else { std::cout << "get tiling func failed." << std::endl; return -1; } // ... |
父主题: OpTilingRegistry