chore: import upstream snapshot with attribution

paddle/phi/api/include/tensor_utils.h

@@ -0,0 +1,84 @@
+/* Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License. */
+
+#pragma once
+
+#include <functional>
+
+#include "paddle/phi/api/include/tensor.h"
+#ifdef PADDLE_WITH_DISTRIBUTE
+#include "paddle/phi/core/distributed/auto_parallel/reshard/reshard_function.h"
+#include "paddle/phi/core/distributed/auto_parallel/reshard/reshard_function_registry.h"
+#include "paddle/phi/core/distributed/auto_parallel/reshard/reshard_utils.h"
+#endif
+
+namespace paddle {
+
+using Deleter = std::function<void(void*)>;
+
+/**
+ * @brief Construct a Tensor from a buffer pointed to by `data`
+ *
+ * @note `from_blob` doesn't copy or move data, Modifying the constructed tensor
+ *       is equivalent to modifying the original data.
+ *
+ * @param data The pointer to the memory buffer.
+ * @param shape The dims of the tensor.
+ * @param strides The strides of the tensor.
+ * @param dtype The data type of the tensor, should correspond to data type of
+ *              `data`. See PD_FOR_EACH_DATA_TYPE in `phi/common/data_type.h`
+ * @param layout The data layout of the tensor.
+ * @param place The place where the tensor is located.
+ *              If `place` is default value, it will be inferred from `data`,
+ *              However, the feature is only supported on CPU or GPU.
+ *              If `place` is not default value, make sure that `place` is equal
+ *              to the place of `data`
+ * @param deleter A function or function object that will be called to free the
+ *                memory buffer.
+ *
+ * @return A Tensor object constructed from the buffer
+ */
+PADDLE_API Tensor from_blob(void* data,
+                            const phi::IntArray& shape,
+                            phi::DataType dtype,
+                            phi::DataLayout layout = phi::DataLayout::NCHW,
+                            const phi::Place& place = phi::Place(),
+                            const Deleter& deleter = nullptr);
+PADDLE_API Tensor from_blob(void* data,
+                            const phi::IntArray& shape,
+                            const phi::IntArray& strides,
+                            phi::DataType dtype,
+                            phi::DataLayout layout = phi::DataLayout::NCHW,
+                            const phi::Place& place = phi::Place(),
+                            const Deleter& deleter = nullptr);
+
+#ifdef PADDLE_WITH_DISTRIBUTE
+/**
+ * @brief Reshard a DistTensor by given DistAttr.
+ *
+ * @note Input of `Reshard` should be a `paddle::Tensor` whose impl is
+ * shared_ptr of DistTensor. According to the given DistAttr, input will be
+ * reshard to wanted distributed state. And it will return shared_ptr of a new
+ * DistTensor as output.
+ *
+ * @param input The input tensor to be resharded.
+ * @param dist_attr The dist_attr to be resharded.
+ * @return Shared_ptr of a new DistTensor
+ */
+// TODO(GhostScreaming): All APIs should call this unified function later.
+PADDLE_API std::shared_ptr<phi::distributed::DistTensor> reshard(
+    const paddle::Tensor& input,
+    const phi::distributed::TensorDistAttr& dist_attr);
+#endif
+}  // namespace paddle