Export some symbols on Windows (#70443)

0x3878f · web-flow · commit f7b4acc893fd · 2024-12-25T20:50:02.000+08:00
diff --git a/paddle/fluid/ir_adaptor/translator/op_compat_info.h b/paddle/fluid/ir_adaptor/translator/op_compat_info.h
@@ -19,6 +19,7 @@
 #include <unordered_set>
 
 #include "glog/logging.h"
+#include "paddle/pir/include/core/dll_decl.h"
 
 #pragma once
 
@@ -30,7 +31,7 @@ using MutableAttributeInfo = std::vector<std::string>;
 static constexpr char kPhiGradSuffix[] = "_grad";
 static constexpr char kFluidVarGradSuffix[] = "@GRAD";
 
-class OpNameNormalizer {
+class IR_API OpNameNormalizer {
  private:
   OpNameNormalizer();  // Disallow instantiation outside of the class.
   std::unordered_map<std::string, std::string> op_name_mappings;
diff --git a/paddle/fluid/pir/dialect/operator/ir/control_flow_op.h b/paddle/fluid/pir/dialect/operator/ir/control_flow_op.h
@@ -19,31 +19,33 @@
 #include "paddle/fluid/pir/dialect/operator/interface/op_yaml_info.h"
 #include "paddle/fluid/pir/dialect/operator/interface/vjp.h"
 #include "paddle/pir/include/core/block.h"
+#include "paddle/pir/include/core/dll_decl.h"
 #include "paddle/pir/include/core/op_base.h"
 
 namespace paddle {
 namespace dialect {
 
-class IfOp : public pir::Op<IfOp, VjpInterface, InferSymbolicShapeInterface> {
+class IR_API IfOp
+    : public pir::Op<IfOp, VjpInterface, InferSymbolicShapeInterface> {
  public:
   using Op::Op;
   static const char *name() { return "pd_op.if"; }
   static constexpr const char **attributes_name = nullptr;
   static constexpr uint32_t attributes_num = 0;
-  TEST_API static void Build(pir::Builder &builder,             // NOLINT
-                             pir::OperationArgument &argument,  // NOLINT
-                             pir::Value cond,
-                             std::vector<pir::Type> &&output_types);
+  static void Build(pir::Builder &builder,             // NOLINT
+                    pir::OperationArgument &argument,  // NOLINT
+                    pir::Value cond,
+                    std::vector<pir::Type> &&output_types);
 
-  TEST_API static void Build(pir::Builder &builder,             // NOLINT
-                             pir::OperationArgument &argument,  // NOLINT
-                             pir::Value cond,
-                             std::unique_ptr<pir::Block> &&true_block,
-                             std::unique_ptr<pir::Block> &&false_block);
+  static void Build(pir::Builder &builder,             // NOLINT
+                    pir::OperationArgument &argument,  // NOLINT
+                    pir::Value cond,
+                    std::unique_ptr<pir::Block> &&true_block,
+                    std::unique_ptr<pir::Block> &&false_block);
 
   pir::Value cond() { return operand_source(0); }
-  TEST_API pir::Block &true_block();
-  TEST_API pir::Block &false_block();
+  pir::Block &true_block();
+  pir::Block &false_block();
   pir::Region &true_region() { return (*this)->region(0); }
   pir::Region &false_region() { return (*this)->region(1); }
   void Print(pir::IrPrinter &printer);  // NOLINT
@@ -70,21 +72,21 @@ class IfOp : public pir::Op<IfOp, VjpInterface, InferSymbolicShapeInterface> {
 ///      cond, outputs = body(outputs)
 ///   }
 ///
-class WhileOp
+class IR_API WhileOp
     : public pir::Op<WhileOp, VjpInterface, InferSymbolicShapeInterface> {
  public:
   using Op::Op;
   static const char *name() { return "pd_op.while"; }
   static constexpr uint32_t attributes_num = 0;
   static constexpr const char **attributes_name = nullptr;
 
-  TEST_API static void Build(pir::Builder &builder,             // NOLINT
-                             pir::OperationArgument &argument,  // NOLINT
-                             pir::Value cond,
-                             const std::vector<pir::Value> &inputs,
-                             bool construct_body = true);
-  TEST_API pir::Block &body();
-  TEST_API pir::Value cond();
+  static void Build(pir::Builder &builder,             // NOLINT
+                    pir::OperationArgument &argument,  // NOLINT
+                    pir::Value cond,
+                    const std::vector<pir::Value> &inputs,
+                    bool construct_body = true);
+  pir::Block &body();
+  pir::Value cond();
   const pir::Block::ArgsType &block_args() { return body().args(); }
   void Print(pir::IrPrinter &printer);  // NOLINT
   void VerifySig();
diff --git a/paddle/fluid/pir/dialect/operator/ir/op_attribute.h b/paddle/fluid/pir/dialect/operator/ir/op_attribute.h
@@ -19,6 +19,7 @@
 #include "paddle/phi/common/scalar.h"
 #include "paddle/phi/core/enforce.h"
 #include "paddle/pir/include/core/builtin_attribute.h"
+#include "paddle/pir/include/core/dll_decl.h"
 #include "paddle/pir/include/core/parser/ir_parser.h"
 
 namespace paddle {
@@ -27,7 +28,7 @@ namespace dialect {
 inline const char kForceBackendAttr[] = "__force_backend__";
 inline const char kCanRunTrtAttr[] = "__l_trt__";
 
-class IntArrayAttribute : public pir::Attribute {
+class IR_API IntArrayAttribute : public pir::Attribute {
  public:
   using Attribute::Attribute;
 
@@ -70,7 +71,7 @@ class ScalarAttribute : public pir::Attribute {
   static std::string name() { return "a_scalar"; }
 };
 
-class DataTypeAttribute : public pir::Attribute {
+class IR_API DataTypeAttribute : public pir::Attribute {
  public:
   using Attribute::Attribute;
 
diff --git a/paddle/fluid/pir/dialect/operator/ir/op_type.h b/paddle/fluid/pir/dialect/operator/ir/op_type.h
@@ -17,6 +17,7 @@
 #include "paddle/fluid/pir/dialect/operator/ir/type_storage.h"
 #include "paddle/pir/include/core/builtin_type.h"
 #include "paddle/pir/include/core/builtin_type_interfaces.h"
+#include "paddle/pir/include/core/dll_decl.h"
 #include "paddle/pir/include/core/type.h"
 #include "paddle/utils/test_macros.h"
 
@@ -63,7 +64,7 @@ class TEST_API SelectedRowsType
   }
 };
 
-class DenseTensorArrayType
+class IR_API DenseTensorArrayType
     : public pir::Type::TypeBase<DenseTensorArrayType,
                                  pir::Type,
                                  DenseTensorArrayTypeStorage> {
diff --git a/paddle/fluid/pir/dialect/operator/utils/op_yaml_info_parser.h b/paddle/fluid/pir/dialect/operator/utils/op_yaml_info_parser.h
@@ -15,6 +15,7 @@
 #pragma once
 
 #include "paddle/fluid/pir/dialect/operator/interface/op_yaml_info.h"
+#include "paddle/pir/include/core/dll_decl.h"
 
 namespace paddle {
 namespace dialect {
@@ -35,8 +36,8 @@ class OpYamlInfoParser {
   const std::vector<std::string>& TensorParams(bool is_kernel = false) const;
   const std::vector<std::string>& AttrParams(bool is_kernel = false) const;
   const OpRunTimeInfo& OpRuntimeInfo() const;
-  const std::map<std::string, uint32_t>& InputName2Id() const;
-  const std::map<std::string, uint32_t>& OutputName2Id() const;
+  TEST_API const std::map<std::string, uint32_t>& InputName2Id() const;
+  TEST_API const std::map<std::string, uint32_t>& OutputName2Id() const;
 
   const std::vector<uint32_t>& NoNeedBufferIds() const;
 
