fix errors in lod_reset_op

scopeFix
fengjiayi 7 years ago
parent e309f42293
commit 2e617334eb

@ -46,8 +46,7 @@ class LoDResetKernel : public framework::OpKernel<T> {
auto* lod = lod_t->data<int>(); auto* lod = lod_t->data<int>();
if (platform::is_gpu_place(ctx.GetPlace())) { if (platform::is_gpu_place(ctx.GetPlace())) {
framework::Tensor lod_cpu; framework::Tensor lod_cpu;
framework::TensorCopy(*lod_t, platform::CPUPlace(), framework::TensorCopySync(*lod_t, platform::CPUPlace(), &lod_cpu);
ctx.device_context(), &lod_cpu);
lod = lod_cpu.data<int>(); lod = lod_cpu.data<int>();
} }
level0 = std::vector<int>(lod, lod + lod_t->numel()); level0 = std::vector<int>(lod, lod + lod_t->numel());

Loading…
Cancel
Save