MindSpore自定义算子报错ERROR: No kernel found for [MyCustomOp] in device GPU

[chengxiaoli (Cheng_li)]

1 系统环境

硬件环境(Ascend/GPU/CPU): GPU/CPU
MindSpore版本: MindSpore=2.6.0
执行模式（PyNative/ Graph）: 不限
Python版本: Python=3.8
操作系统平台: linux

2 报错信息

2.1 问题描述

按照 MindSpore 的规范，成功实现了一个自定义算子（一个名为MyCustomOp的 element-wise 操作），并且在 CPU 后端上能够正常编译和运行。

2.2 报错信息

然而当尝试切换到 GPU 后端，通过如下设置

context.set_context(device_target="GPU")

并重新编译运行时，程序会报错并退出，错误信息类似于：

ERROR: mindspore/ccsrc/backend/kernel_compiler/gpu/gpu_kernel_factory.cc:56] No kernel found for [MyCustomOp] in device GPU

3 根因分析

• 此处由用户填写

4 解决方案

• 此处由用户填写
• 包含文字方案和最终脚本代码 请将正确的脚本打包并上传附件