CANN/metadef算子简化键生成
GenSimplifiedKey【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能注册算子的GenSimplifiedKey函数以提供一个更加快速的二进制匹配key值。用户需要为算子编写一个GenSimplifiedKey类型的函数并使用该接口进行注册。GenSimplifiedKey类型定义如下using GenSimplifiedKeyKernelFunc UINT32 (*)(TilingContext *, ge::char_t *);GenSimplifiedKey函数接受一个TilingContext类型参数和ge::char_t类型参数作为输入通过该输入算子可自定义simplified key生成逻辑。函数原型OpImplRegisterV2 GenSimplifiedKey(GenSimplifiedKeyKernelFunc gen_simplifiedkey_func)参数说明参数输入/输出说明gen_simplifiedkey_func输入要注册的自定义GenSimplifiedKey函数类型为GenSimplifiedKeyKernelFunc。GenSimplifiedKeyKernelFunc类型定义如下using GenSimplifiedKeyKernelFunc UINT32 (*)(TilingContext *, ge::char_t *);返回值说明返回算子的OpImplRegisterV2对象该对象新增注册了生成二进制简化匹配key函数。约束说明无。【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考