GetTilingFunc

功 能 说 明

根 据 算 子 类 型OpType获 取 对 应 的Tiling函 数。

函 数 原 型

Text

TilingFunc GetTilingFunc(const char *opType) const

参 数 说 明

参 数 名	输 入/输 出	描 述
opType	输 入	算 子 类 型，与 原 型 定 义 时 的 算 子 类 型 保 持 一 致。

返 回 值 说 明

获 取 成 功，则 返 回 对 应 的Tiling函 数 指 针；失 败 则 返 回 空 指 针。Tiling函 数 指 针 定 义 如 下：

Text

using TilingFunc = uint32_t(*)(gert::TilingContext*)

约 束 说 明

无

调 用 示 例

Text

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;
}
// ...