首页 Paddle框架 帖子详情
定义外部算子 atomic_add有无接口?
收藏
快速回复
Paddle框架 问答模型训练 1064 1
定义外部算子 atomic_add有无接口?
收藏
快速回复
Paddle框架 问答模型训练 1064 1

c++定义的外部算子 需要用cuda的atmoic_add paddle有无接口

还是可以直接使用,要包含什么头文件吗?不太懂这块

0
收藏
回复
全部评论(1)
时间顺序
TC.Long
#2 回复于2021-06

使用C++自定义外部算子,可以参考下官网关于自定义算子的内容:https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/07_new_op/new_custom_op_cn.html

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户