From a2b6fe2deab9b02fe76191161cf01320390ced54 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?=
 <83450930+Liyulingyue@users.noreply.github.com>
Date: Sat, 9 Mar 2024 05:24:43 +0800
Subject: [PATCH] Update op_utils.h

---
 paddle/phi/core/compat/op_utils.h | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/paddle/phi/core/compat/op_utils.h b/paddle/phi/core/compat/op_utils.h
index 0e2719d950406..515b67f4d42a1 100644
--- a/paddle/phi/core/compat/op_utils.h
+++ b/paddle/phi/core/compat/op_utils.h
@@ -243,6 +243,7 @@ struct ArgumentMappingFnRegistrar {
       __registrar_arg_map_fn_for_##op_type(#op_type, arg_mapping_fn);    \
   int TouchArgumentMappingFnSymbol_##op_type() { return 0; }
 
+#if(WIN32)
 #define PD_DECLARE_ARG_MAPPING_FN(op_type)                              \
   PD_STATIC_ASSERT_GLOBAL_NAMESPACE(                                    \
       PD_DECLARE_arg_map_fn_ns_check_##op_type,                         \
@@ -250,5 +251,13 @@ struct ArgumentMappingFnRegistrar {
   TEST_API extern int TouchArgumentMappingFnSymbol_##op_type();         \
   UNUSED static int __declare_arg_map_fn_symbol_for_##op_type =         \
       TouchArgumentMappingFnSymbol_##op_type()
-
+#else
+#define PD_DECLARE_ARG_MAPPING_FN(op_type)                              \
+  PD_STATIC_ASSERT_GLOBAL_NAMESPACE(                                    \
+      PD_DECLARE_arg_map_fn_ns_check_##op_type,                         \
+      "PD_DECLARE_ARG_MAPPING_FN must be called in global namespace."); \
+  __declspec(dllexport) TEST_API extern int TouchArgumentMappingFnSymbol_##op_type();         \
+  UNUSED static int __declare_arg_map_fn_symbol_for_##op_type =         \
+      TouchArgumentMappingFnSymbol_##op_type()
+#endif
 }  // namespace phi