GetTilingFunc
功能说明
根据算子类型OpType获取对应的Tiling函数。
函数原型
TilingFunc GetTilingFunc(const char *opType) const
参数说明
返回值说明
获取成功,则返回对应的Tiling函数指针;失败则返回空指针。Tiling函数指针定义如下:
using TilingFunc = uint32_t(*)(gert::TilingContext*)
约束说明
无
调用示例
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;
}
// ...