call resize directly in constructor

JiabinYang · Mar 24, 2021 · 36e4b49 · 36e4b49
1 parent 1384322
commit 36e4b49
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/extension/src/ext_tensor.cc b/paddle/fluid/extension/src/ext_tensor.cc
@@ -119,7 +119,8 @@ Tensor::Tensor(const PlaceType &place, const std::vector<int64_t> &shape)
     : tensor_(std::make_shared<framework::LoDTensor>()),
       place_(place),
       stream_(StreamWrapper()) {
-  reshape(shape);
+  GET_CASTED_TENSOR
+  tensor->Resize(framework::make_ddim(shape));
 }
 
 template <typename T>