Merge pull request #10548 from JiayiFeng/fix_warpctc_op

fix warpctc
8 years ago · 718f63c0d2
parent 86d7cc9789 0d10514d4b
commit 718f63c0d2
1 changed files with 1 additions and 2 deletions
--- a/paddle/fluid/operators/warpctc_op.h
+++ b/paddle/fluid/operators/warpctc_op.h
@ -186,8 +186,7 @@ class WarpCTCKernel : public framework::OpKernel<T> {

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