1
模型应用
标题
2025-07-08 10:48:59
59

请教,migraphx支持自定义算子吗,可以的话大概流程怎么实现,是否有基于dtk的MIGraphX 4.1.0源代码

环境版本:docker中,卡Z100L 2个,Ubuntu20.04.1,dtk23.10.1,MIGraphX version: 4.1.0

全部评论(3)
只看楼主
网
网友bX8AjM
1楼
您好,目前migraphx自定义算子还处于实验阶段,功能尚不完善。临时方案可以参考官方示例工程:https://github.com/ROCm/AMDMIGraphX/blob/develop/examples/migraphx/custom_op_hip_kernel/custom_op_hip_kernel.cpp
2025/7/8
网
网友8c8qbv
回复 @网友bX8AjM
谢谢,migraphx源码添加算子后直接编译就行吗,onnxruntime支持自定义算子吗,添加算子源码编译可行不
2025/7/9
网
网友8c8qbv
回复 @网友bX8AjM
@网友bX8AjM 请教以我的环境是否行得通,您有什么建议吗
2025/7/10
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据