【paddle_test No.21】replace parts of cc_test with paddle_test (#61674)

Liyulingyue · web-flow · commit 8ed00b690ff1 · 2024-03-05T19:43:02.000+08:00
* Update CMakeLists.txt

* add TEST_API

* fix code-style

* add test_API

* Apply suggestions from code review

* Update CMakeLists.txt

* modify CMakeLists.txt

* add TESTAPI

* add .h file to import TEST_API

* Update lod_utils.h

* add parts of TEST_API

* Apply suggestions from code review

* add TEST_API

* Apply suggestions from code review

* Apply suggestions from code review

* add TEST_API
diff --git a/paddle/fluid/framework/attribute.h b/paddle/fluid/framework/attribute.h
@@ -34,9 +34,9 @@ limitations under the License. */
 namespace paddle {
 namespace framework {
 
-paddle::any GetAttrValue(const Attribute& attr);
+TEST_API paddle::any GetAttrValue(const Attribute& attr);
 
-Attribute GetAttrValue(const proto::OpDesc::Attr& attr_desc);
+TEST_API Attribute GetAttrValue(const proto::OpDesc::Attr& attr_desc);
 
 Attribute GetAttrValue(const proto::VarDesc::Attr& attr_desc);
 
@@ -350,9 +350,10 @@ class AttrReader {
 };
 
 paddle::experimental::Scalar MakeScalarFromProto(const proto::Scalar& v);
-proto::Scalar MakeScalarProto(const paddle::experimental::Scalar& v);
-paddle::experimental::Scalar MakeScalarFromAttribute(const Attribute& v);
-std::vector<paddle::experimental::Scalar> MakeScalarsFromAttribute(
+TEST_API proto::Scalar MakeScalarProto(const paddle::experimental::Scalar& v);
+TEST_API paddle::experimental::Scalar MakeScalarFromAttribute(
+    const Attribute& v);
+TEST_API std::vector<paddle::experimental::Scalar> MakeScalarsFromAttribute(
     const Attribute& v);
 void CanonicalizeScalarAttrs(const proto::OpProto& op_proto,
                              AttributeMap* attrs);
diff --git a/paddle/fluid/framework/data_layout_transform.h b/paddle/fluid/framework/data_layout_transform.h
@@ -47,11 +47,11 @@ struct CastDataLayout {
 
 std::vector<int> GetAxis(const DataLayout& from, const DataLayout& to);
 
-void TransDataLayout(const phi::KernelKey& kernel_type_for_var,
-                     const phi::KernelKey& expected_kernel_type,
-                     const phi::DenseTensor& in,
-                     phi::DenseTensor* out,
-                     const phi::Place& place);
+TEST_API void TransDataLayout(const phi::KernelKey& kernel_type_for_var,
+                              const phi::KernelKey& expected_kernel_type,
+                              const phi::DenseTensor& in,
+                              phi::DenseTensor* out,
+                              const phi::Place& place);
 
 void TransDataLayout(phi::DataLayout from_layout,
                      phi::DataLayout to_layout,
diff --git a/paddle/fluid/framework/data_type.h b/paddle/fluid/framework/data_type.h
@@ -29,7 +29,7 @@ namespace paddle {
 namespace framework {
 
 TEST_API std::string DataTypeToString(const proto::VarType::Type type);
-extern size_t SizeOfType(proto::VarType::Type type);
+TEST_API extern size_t SizeOfType(proto::VarType::Type type);
 
 template <typename T>
 struct IsComplex : public std::false_type {};
@@ -123,7 +123,7 @@ _ForEachDataType_(DefineDataTypeTrait);
 
 #undef DefineDataTypeTrait
 
-extern proto::VarType::Type ToDataType(std::type_index type);
+TEST_API extern proto::VarType::Type ToDataType(std::type_index type);
 extern std::type_index ToTypeIndex(proto::VarType::Type type);
 
 template <typename Visitor>
diff --git a/paddle/fluid/framework/data_type_transform.h b/paddle/fluid/framework/data_type_transform.h
@@ -28,10 +28,10 @@ class OpKernelType;
 
 using KernelTypePair = std::pair<OpKernelType, OpKernelType>;
 
-void TransDataType(const phi::KernelKey& kernel_type_for_var,
-                   const phi::KernelKey& expected_kernel_type,
-                   const phi::DenseTensor& in,
-                   phi::DenseTensor* out);
+TEST_API void TransDataType(const phi::KernelKey& kernel_type_for_var,
+                            const phi::KernelKey& expected_kernel_type,
+                            const phi::DenseTensor& in,
+                            phi::DenseTensor* out);
 void TransDataType(const phi::DenseTensor& in,
                    const paddle::framework::proto::VarType::Type& type,
                    phi::DenseTensor* out);
diff --git a/paddle/fluid/framework/device_worker.h b/paddle/fluid/framework/device_worker.h
@@ -60,20 +60,21 @@ class Scope;
 namespace paddle {
 namespace framework {
 
-std::string PrintLodTensor(phi::DenseTensor* tensor,
-                           int64_t start,
-                           int64_t end,
-                           char separator = ',',
-                           bool need_leading_separator = false);
-void PrintLodTensor(phi::DenseTensor* tensor,
-                    int64_t start,
-                    int64_t end,
-                    std::string& output_str,  // NOLINT
-                    char separator = ',',
-                    bool need_leading_separator = false,
-                    int num_decimals = 9);
-std::pair<int64_t, int64_t> GetTensorBound(phi::DenseTensor* tensor, int index);
-bool CheckValidOutput(phi::DenseTensor* tensor, size_t batch_size);
+TEST_API std::string PrintLodTensor(phi::DenseTensor* tensor,
+                                    int64_t start,
+                                    int64_t end,
+                                    char separator = ',',
+                                    bool need_leading_separator = false);
+TEST_API void PrintLodTensor(phi::DenseTensor* tensor,
+                             int64_t start,
+                             int64_t end,
+                             std::string& output_str,  // NOLINT
+                             char separator = ',',
+                             bool need_leading_separator = false,
+                             int num_decimals = 9);
+TEST_API std::pair<int64_t, int64_t> GetTensorBound(phi::DenseTensor* tensor,
+                                                    int index);
+TEST_API bool CheckValidOutput(phi::DenseTensor* tensor, size_t batch_size);
 
 class FleetWrapper;
 
diff --git a/paddle/fluid/framework/dlpack_tensor.h b/paddle/fluid/framework/dlpack_tensor.h
@@ -28,7 +28,8 @@ class DLPackTensor {
       std::remove_reference<decltype(::DLTensor::shape[0])>::type;  // int64_t
 
   // lanes is only used in CPU to enable vectorization
-  explicit DLPackTensor(const phi::DenseTensor& tensor, LaneType lanes = 1);
+  TEST_API explicit DLPackTensor(const phi::DenseTensor& tensor,
+                                 LaneType lanes = 1);
 
   inline operator const ::DLTensor&() const { return t_; }
 
diff --git a/paddle/fluid/framework/lod_tensor.h b/paddle/fluid/framework/lod_tensor.h
@@ -27,17 +27,19 @@ limitations under the License. */
 #include "paddle/fluid/platform/place.h"
 #include "paddle/phi/core/dense_tensor.h"
 #include "paddle/phi/core/mixed_vector.h"
+#include "paddle/utils/test_macros.h"
 
 namespace paddle {
 namespace framework {
 
 // Split phi::DenseTensor and copy to each place specified in places.
-std::vector<phi::DenseTensor> SplitLoDTensor(
+TEST_API std::vector<phi::DenseTensor> SplitLoDTensor(
     const phi::DenseTensor& src, const std::vector<platform::Place> places);
 
-void MergeLoDTensor(phi::DenseTensor* target,
-                    const std::vector<const phi::DenseTensor*>& lod_tensors,
-                    platform::Place dst_place);
+TEST_API void MergeLoDTensor(
+    phi::DenseTensor* target,
+    const std::vector<const phi::DenseTensor*>& lod_tensors,
+    platform::Place dst_place);
 
 /*
  * LoD is short for Level of Details.
@@ -65,7 +67,7 @@ LoD SliceInLevel(const LoD& in,
 /*
  * Transform an LoD from relative offsets to absolute offsets.
  */
-LoD ToAbsOffset(const LoD& in);
+TEST_API LoD ToAbsOffset(const LoD& in);
 
 TEST_API bool operator==(const LoD& a, const LoD& b);
 
@@ -85,7 +87,7 @@ TEST_API bool operator==(const LoD& a, const LoD& b);
  * tensor_height>0.
  */
 
-bool CheckLoD(const LoD& in, int tensor_height = -1);
+TEST_API bool CheckLoD(const LoD& in, int tensor_height = -1);
 /*
  * Check whether this absolute lod's format is valid.
  *
@@ -99,7 +101,7 @@ bool CheckLoD(const LoD& in, int tensor_height = -1);
  *     same(the height of underlying tensor) or `tensor_height` if
  *     tensor_height>0.
  */
-bool CheckAbsLoD(const LoD& in, int tensor_height = -1);
+TEST_API bool CheckAbsLoD(const LoD& in, int tensor_height = -1);
 
 /*
  * Expand the `source` to fit the LoD of `lod`. For example, a `source`
@@ -162,7 +164,7 @@ phi::DenseTensor LodExpand(const phi::DenseTensor& source,
 // Returns:
 //  LoD = [[1, 4], [2, 4, 2, 3, 2]]
 //  pair<size_t, size_t> = {11, 24}
-std::pair<LoD, std::pair<size_t, size_t>> GetSubLoDAndAbsoluteOffset(
+TEST_API std::pair<LoD, std::pair<size_t, size_t>> GetSubLoDAndAbsoluteOffset(
     const LoD& lod, size_t start_idx, size_t end_idx, size_t start_level);
 
 /*
@@ -182,7 +184,7 @@ void DeserializeFromStream(std::istream& is,
                            const size_t& seek,
                            const std::vector<int64_t>& shape);
 
-LoD ConvertToOffsetBasedLoD(const LoD& length_lod);
+TEST_API LoD ConvertToOffsetBasedLoD(const LoD& length_lod);
 
 void SerializeToStream(std::ostream& os, const phi::DenseTensor& tensor);
 
diff --git a/paddle/fluid/framework/reader.h b/paddle/fluid/framework/reader.h
@@ -48,15 +48,15 @@ class ReaderBase {
             "and need_check_feed"));
   }
 
-  virtual void ReadNext(paddle::framework::LoDTensorArray* out);
+  TEST_API virtual void ReadNext(paddle::framework::LoDTensorArray* out);
 
-  virtual void Shutdown();
+  TEST_API virtual void Shutdown();
 
-  virtual void Start();
+  TEST_API virtual void Start();
 
   // Return the readers which are the end of decorating chain. Basically
   // they are readers just before read op.
-  std::unordered_set<ReaderBase*> GetEndPoints();
+  TEST_API std::unordered_set<ReaderBase*> GetEndPoints();
 
   // Returns the shapes of the fed variables
   const std::vector<DDim>& Shapes() const { return shapes_; }
@@ -70,7 +70,7 @@ class ReaderBase {
   // This function returns whether you have the check shape for this Reader.
   const std::vector<bool>& NeedCheckFeed() const { return need_check_feed_; }
 
-  virtual ~ReaderBase();
+  TEST_API virtual ~ReaderBase();
 
  protected:
   virtual void ReadNextImpl(paddle::framework::LoDTensorArray* out UNUSED) {}
@@ -98,7 +98,7 @@ class ReaderBase {
   friend class DecoratedReader;
   // These methods can be only invoked inside DecoratedReader to record the
   // decorating chain.
-  void InsertDecoratedReader(
+  TEST_API void InsertDecoratedReader(
       const std::shared_ptr<ReaderBase>& decorated_reader);
   // A set of which readers that decorated this reader.
   std::vector<std::weak_ptr<ReaderBase>> decorated_readers_;
@@ -121,7 +121,7 @@ class DecoratedReader : public ReaderBase,
     reader_->InsertDecoratedReader(shared_from_this());
   }
 
-  ~DecoratedReader();
+  TEST_API ~DecoratedReader();
 
   const std::shared_ptr<ReaderBase>& UnderlyingReader() const {
     return reader_;
diff --git a/paddle/fluid/framework/tensor_util.h b/paddle/fluid/framework/tensor_util.h
@@ -53,12 +53,12 @@ class PrintOptions {
   PrintOptions() {}
 };
 
-void TensorToStream(std::ostream& os,
-                    const phi::DenseTensor& tensor,
-                    const platform::DeviceContext& dev_ctx);
-void TensorFromStream(std::istream& is,
-                      phi::DenseTensor* tensor,
-                      const platform::DeviceContext& dev_ctx);
+TEST_API void TensorToStream(std::ostream& os,
+                             const phi::DenseTensor& tensor,
+                             const platform::DeviceContext& dev_ctx);
+TEST_API void TensorFromStream(std::istream& is,
+                               phi::DenseTensor* tensor,
+                               const platform::DeviceContext& dev_ctx);
 void TensorFromStream(std::istream& is,
                       phi::DenseTensor* tensor,
                       const platform::DeviceContext& dev_ctx,
@@ -107,7 +107,8 @@ void TensorToVector(const phi::DenseTensor& src, std::vector<T>* dst);
 
 // convert dlpack's DLTensor to tensor
 
-void TensorFromDLPack(const ::DLTensor& dl_tensor, phi::DenseTensor* dst);
+TEST_API void TensorFromDLPack(const ::DLTensor& dl_tensor,
+                               phi::DenseTensor* dst);
 void TensorFromDLPack(const DLManagedTensor* src, phi::DenseTensor* dst);
 
 //
diff --git a/paddle/fluid/framework/var_type_traits.h b/paddle/fluid/framework/var_type_traits.h
@@ -97,8 +97,8 @@ namespace paddle {
 namespace framework {
 
 TEST_API const char *ToTypeName(int var_id);
-const std::type_index &VarTraitIdToTypeIndex(int var_id);
-int TypeIndexToVarTraitId(const std::type_index &type);
+TEST_API const std::type_index &VarTraitIdToTypeIndex(int var_id);
+TEST_API int TypeIndexToVarTraitId(const std::type_index &type);
 
 namespace detail {
 
diff --git a/paddle/phi/common/scalar.h b/paddle/phi/common/scalar.h
@@ -356,9 +356,9 @@ void CopyScalar(const ScalarBase<T1>& src, ScalarBase<T2>* dst) {
 }
 
 using Scalar = paddle::experimental::ScalarBase<Tensor>;
-bool operator==(const Scalar& lhs, const Scalar& rhs);
+TEST_API bool operator==(const Scalar& lhs, const Scalar& rhs);
 
-std::ostream& operator<<(std::ostream& os, const Scalar& s);
+TEST_API std::ostream& operator<<(std::ostream& os, const Scalar& s);
 
 template <typename T>
 std::vector<T> ExtractPlainVector(
diff --git a/paddle/phi/core/compat/convert_utils.h b/paddle/phi/core/compat/convert_utils.h
@@ -29,7 +29,7 @@ namespace phi {
 const std::string& TransToPhiKernelName(const std::string& fluid_op_name);
 const std::string& TransToFluidOpName(const std::string& phi_kernel_name);
 
-Backend TransToPhiBackend(const phi::Place& place);
+TEST_API Backend TransToPhiBackend(const phi::Place& place);
 phi::Place TransToPhiPlace(const Backend& backend, bool set_device_id = true);
 
 #ifdef PADDLE_WITH_DNNL
diff --git a/paddle/phi/core/lod_utils.h b/paddle/phi/core/lod_utils.h
@@ -16,6 +16,8 @@
 #include <cstddef>
 #include <vector>
 
+#include "paddle/utils/test_macros.h"
+
 namespace phi {
 using LoD = std::vector<std::vector<std::size_t>>;
 
@@ -24,7 +26,7 @@ using LoD = std::vector<std::vector<std::size_t>>;
  */
 LoD ToAbsOffset(const LoD& in);
 
-void AppendLoD(LoD* lod, const LoD& lod_length);
+TEST_API void AppendLoD(LoD* lod, const LoD& lod_length);
 
 /*
  * Convert between length-based LoD and offset-based LoD.
@@ -36,6 +38,6 @@ void AppendLoD(LoD* lod, const LoD& lod_length);
  * If offset_lod = [[0, 2, 3],[0, 3, 5, 9]]
  * then length_lod = [[2, 1], [3, 2, 4]]
  */
-LoD ConvertToLengthBasedLoD(const LoD& offset_lod);
+TEST_API LoD ConvertToLengthBasedLoD(const LoD& offset_lod);
 
 }  // namespace  phi
diff --git a/paddle/phi/core/tensor_utils.h b/paddle/phi/core/tensor_utils.h
@@ -134,7 +134,8 @@ void TensorToVector(const phi::DenseTensor& src,
                     const phi::DeviceContext& ctx,
                     std::vector<T>* dst);
 
-phi::DenseTensor ReshapeToMatrix(const phi::DenseTensor& src, int num_col_dims);
+TEST_API phi::DenseTensor ReshapeToMatrix(const phi::DenseTensor& src,
+                                          int num_col_dims);
 
 template <typename T>
 T GetValue(const phi::DenseTensor* x);
diff --git a/paddle/phi/core/threadpool.h b/paddle/phi/core/threadpool.h
@@ -56,7 +56,7 @@ class ThreadPool {
       std::packaged_task<std::unique_ptr<common::enforce::EnforceNotMet>()>;
 
   // Returns the singleton of ThreadPool.
-  static ThreadPool* GetInstance();
+  TEST_API static ThreadPool* GetInstance();
 
   ~ThreadPool();
 
diff --git a/paddle/phi/kernels/funcs/data_layout_transform.h b/paddle/phi/kernels/funcs/data_layout_transform.h
@@ -83,7 +83,8 @@ void TransDataLayoutFromOneDNN(DataLayout in_layout,
                                DenseTensor* out,
                                Place place,
                                bool always_copy = false);
-void* GetDataFromTensor(const DenseTensor& tensor, OneDNNDataType type);
+TEST_API void* GetDataFromTensor(const DenseTensor& tensor,
+                                 OneDNNDataType type);
 
 dnnl::memory::desc make_memory_desc(const phi::DenseTensor& ref_tensor,
                                     phi::DataLayout target_layout);
diff --git a/paddle/phi/kernels/isfinite_kernel.h b/paddle/phi/kernels/isfinite_kernel.h
@@ -20,7 +20,7 @@ namespace phi {
 
 #define DEFINE_ISFINITE_KERNEL(isfinite_kernel) \
   template <typename T, typename Context>       \
-  void isfinite_kernel(                         \
+  TEST_API void isfinite_kernel(                \
       const Context& ctx, const DenseTensor& x, DenseTensor* out);
 
 DEFINE_ISFINITE_KERNEL(IsinfKernel)
diff --git a/paddle/phi/kernels/reduce_all_kernel.h b/paddle/phi/kernels/reduce_all_kernel.h
@@ -27,10 +27,10 @@ void AllRawKernel(const Context& dev_ctx,
                   DenseTensor* out);
 
 template <typename T, typename Context>
-void AllKernel(const Context& dev_ctx,
-               const DenseTensor& x,
-               const std::vector<int64_t>& dims,
-               bool keep_dim,
-               DenseTensor* out);
+TEST_API void AllKernel(const Context& dev_ctx,
+                        const DenseTensor& x,
+                        const std::vector<int64_t>& dims,
+                        bool keep_dim,
+                        DenseTensor* out);
 
 }  // namespace phi
diff --git a/paddle/phi/kernels/reduce_any_kernel.h b/paddle/phi/kernels/reduce_any_kernel.h
@@ -26,10 +26,10 @@ void AnyRawKernel(const Context& dev_ctx,
                   DenseTensor* out);
 
 template <typename T, typename Context>
-void AnyKernel(const Context& dev_ctx,
-               const DenseTensor& x,
-               const std::vector<int64_t>& dims,
-               bool keep_dim,
-               DenseTensor* out);
+TEST_API void AnyKernel(const Context& dev_ctx,
+                        const DenseTensor& x,
+                        const std::vector<int64_t>& dims,
+                        bool keep_dim,
+                        DenseTensor* out);
 
 }  // namespace phi
diff --git a/test/cpp/fluid/framework/CMakeLists.txt b/test/cpp/fluid/framework/CMakeLists.txt
