Skip to content

Commit cb4083b

Browse files
committed
fix compile error
test=develop
1 parent dd343a4 commit cb4083b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/operators/math/fc_compute.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ inline void FCCompute(const BlasT<DeviceContext, T>& blas, const int M,
3636
.template Get<jitkernel::VAddReluKernel<T>>(N);
3737
for (int i = 0; i < M; i++) {
3838
T* dst = Y + i * N;
39-
vaddrelu->Compute(B, dst, dst);
39+
vaddrelu->Compute(B, dst, dst, N);
4040
}
4141
} else {
4242
const auto& vadd = jitkernel::KernelPool::Instance()
@@ -47,7 +47,7 @@ inline void FCCompute(const BlasT<DeviceContext, T>& blas, const int M,
4747
#endif
4848
for (int i = 0; i < M; i++) {
4949
T* dst = Y + i * N;
50-
vadd->Compute(B, dst, dst);
50+
vadd->Compute(B, dst, dst, N);
5151
}
5252
}
5353
}

0 commit comments

Comments
 (0)