Skip to content

Commit 2024489

Browse files
committed
Fix CI
1 parent 2b9b6c3 commit 2024489

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/operators/compare_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ class CompareOpKernel
5454
public:
5555
void Compute(const framework::ExecutionContext& context) const override {
5656
using T = typename Functor::ELEM_TYPE;
57-
ElementwiseComputeEx<Functor, DeviceContext, T>(context);
57+
ElementwiseComputeEx<Functor, DeviceContext, T, bool>(context);
5858
}
5959
};
6060

paddle/operators/elementwise_op_function.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@ void ElementwiseComputeEx(const framework::ExecutionContext& ctx) {
373373
auto* x = ctx.Input<Tensor>("X");
374374
auto* y = ctx.Input<Tensor>("Y");
375375
auto* z = ctx.Output<Tensor>("Out");
376-
z->mutable_data<T>(ctx.GetPlace());
376+
z->mutable_data<OutType>(ctx.GetPlace());
377377
TransformFunctor<Functor, T, DeviceContext, OutType> functor(
378378
x, y, z, ctx.template device_context<DeviceContext>(), Functor());
379379

0 commit comments

Comments
 (0)