We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e309f42 commit 2e61733Copy full SHA for 2e61733
paddle/fluid/operators/lod_reset_op.h
@@ -46,8 +46,7 @@ class LoDResetKernel : public framework::OpKernel<T> {
46
auto* lod = lod_t->data<int>();
47
if (platform::is_gpu_place(ctx.GetPlace())) {
48
framework::Tensor lod_cpu;
49
- framework::TensorCopy(*lod_t, platform::CPUPlace(),
50
- ctx.device_context(), &lod_cpu);
+ framework::TensorCopySync(*lod_t, platform::CPUPlace(), &lod_cpu);
51
lod = lod_cpu.data<int>();
52
}
53
level0 = std::vector<int>(lod, lod + lod_t->numel());
0 commit comments