定义外部算子 atomic_add有无接口?
收藏
c++定义的外部算子 需要用cuda的atmoic_add paddle有无接口
还是可以直接使用,要包含什么头文件吗?不太懂这块
0
收藏
请登录后评论
使用C++自定义外部算子,可以参考下官网关于自定义算子的内容:https://www.paddlepaddle.org.cn/documentation/docs/zh/guides/07_new_op/new_custom_op_cn.html