Skip to content

Commit

Permalink
fix:delete macro INFERENCE (#37130)
Browse files Browse the repository at this point in the history
  • Loading branch information
feng_shuai authored Nov 15, 2021
1 parent df7cc45 commit b628c31
Showing 1 changed file with 0 additions and 7 deletions.
7 changes: 0 additions & 7 deletions paddle/fluid/operators/softmax_op.h
Original file line number Diff line number Diff line change
Expand Up @@ -74,16 +74,9 @@ class SoftmaxKernel : public framework::OpKernel<T> {
Tensor X_2d, Out_2d;
X_2d.ShareDataWith(*X).Resize({n, d});
Out_2d.ShareDataWith(*Out).Resize({n, d});

#ifdef PADDLE_ON_INFERENCE
math::SoftmaxFunctor<DeviceContext, T, true>()(
context.template device_context<DeviceContext>(), axis_dim, &X_2d,
&Out_2d);
#else
math::SoftmaxFunctor<DeviceContext, T, false>()(
context.template device_context<DeviceContext>(), axis_dim, &X_2d,
&Out_2d);
#endif
}
};

Expand Down

0 comments on commit b628c31

Please sign in to comment.