未验证 提交 cecea8e6 编写于 作者: Y Yiqun Liu 提交者: GitHub

Change a comment in pten header to avoid the disturb to op benchmark ci. (#38165)

test=document_fix
上级 f616d9a7
...@@ -130,16 +130,15 @@ void CommonElementwiseBroadcastForward( ...@@ -130,16 +130,15 @@ void CommonElementwiseBroadcastForward(
is_xsize_larger); is_xsize_larger);
} }
// It is a common implementation to compute binary calculation with the support // It is a common CPU implementation to compute binary calculation with the
// of broadcast, supporting both CPU and GPU. // support of broadcast. Note:
// - CPU implementation cannot support the case when x needs broadcast, thus // 1. CPU implementation cannot support the case when x needs broadcast, thus
// this function need to be called with XxxFunctor and XxxInverseFunctor, // this function need to be called with XxxFunctor and XxxInverseFunctor,
// like paddle/fluid/operators/elementwise/elementwise_add_op.h#L49 - L55. // like AddFunctor and InverseAddFunctor.
// - GPU implementation supports all the broadcast cases, thus there is no need // 2. The corresponding GPU implementation supports all the broadcast cases,
// to define and call with XxxInverseFunctor. // thus there is no need to define and call with XxxInverseFunctor.
// TODO(liuyiqun): optimize the CPU implementation to support all broadcast // TODO(liuyiqun): optimize the CPU implementation to support all broadcast
// cases and avoid the need of XxxInverseFunctor. // cases and avoid the need of XxxInverseFunctor.
template <typename Functor, typename T, typename OutType = T> template <typename Functor, typename T, typename OutType = T>
void ElementwiseCompute(const paddle::platform::CPUDeviceContext &dev_ctx, void ElementwiseCompute(const paddle::platform::CPUDeviceContext &dev_ctx,
const DenseTensor &x, const DenseTensor &x,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册