Skip to content

Commit cf3b3d6

Browse files
committed
fix warpctc
1 parent 0285a2b commit cf3b3d6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/fluid/operators/warpctc_op.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -186,8 +186,7 @@ class WarpCTCKernel : public framework::OpKernel<T> {
186186

187187
// warpctc accesses labels in CPU memory
188188
Tensor warpctc_label;
189-
TensorCopy(*label, platform::CPUPlace(), ctx.device_context(),
190-
&warpctc_label);
189+
TensorCopySync(*label, platform::CPUPlace(), &warpctc_label);
191190
const int* warpctc_label_data = warpctc_label.data<int>();
192191
// warpctc stores loss in CPU memory
193192
Tensor warpctc_loss;

0 commit comments

Comments
 (0)