自定义算子内核调用符方式测试通过,API执行报错segmentation fault
收藏回复举报
自定义算子内核调用符方式测试通过,API执行报错segmentation fault
t('forum.solved') 已解决
发表于2024-01-10 14:28:35
0 查看

根据 samples/ cplusplus / level1_single_api / 4_op_dev / 6_ascendc_custom_op / kernel_invocation的内容修改为自定义算子运行成功,且输出符合预期。

但是根据官方文档的步骤进行单算子API执行时在最终./execute_add_op时报错segmentation fault:

请问可能的原因是什么?
为了进行单算子API执行,我在编译部署自定义算子工程后,修改了 samples/ operator / AddCustomSample / FrameworkLaunch / AclNNInvocation路径下的如下文件:

src/CMakeLists:16:set(AUTO_GEN_PATH)
src/main.cpp:(只需要一个输入)修改CreateOpDesc(),SetInputData()
src/op_runner.cpp:276:

OpRunner::RunOp():auto ret = aclnnAddCustomGetWorkspaceSize
OpRunner::RunOp():aclnnAddCustom(workspace, workspaceSize, handle, stream)

scripts/gen_data.py

我要发帖子