Merge pull request #31 from qingshui/paddlebox

fix zero tensor bug
jack603047588 · Feb 16, 2022 · 77c533b · 77c533b
2 parents 78f584c + 9b9307c
commit 77c533b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddle/fluid/framework/data_feed.h b/paddle/fluid/framework/data_feed.h
@@ -1422,6 +1422,7 @@ class MiniBatchGpuPack {
   template <typename T>
   void copy_host2device(Tensor* buf, const T* val, size_t size) {
     if (size == 0) {
+      buf->mutable_data<T>({static_cast<int64_t>(size), 1}, place_);
       return;
     }
     T* data = buf->mutable_data<T>({static_cast<int64_t>(size), 1}, place_);