|
|
@ -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());
|
|
|
|