Skip to content

Commit 718f63c

Browse files
authored
Merge pull request #10548 from JiayiFeng/fix_warpctc_op
fix warpctc
2 parents 86d7cc9 + 0d10514 commit 718f63c

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)