[Prim]check prim forward in windows and mac (#50527)

* check win

* fix random error

* fix manage

python/paddle/fluid/tests/unittests/dygraph_to_static/test_cinn_prim.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import platform
 import unittest
 
 import numpy as np
@@ -134,8 +133,6 @@ class TestPrimForwardAndBackward(unittest.TestCase):
         self.assertTrue('softmax' not in fwd_ops)
 
     def test_cinn_prim(self):
-        plat = platform.system()
-        if plat == "Linux":
         dy_res = self.train(use_prim=False)
         cinn_res = self.train(use_prim=True)
 
@@ -143,8 +140,6 @@ class TestPrimForwardAndBackward(unittest.TestCase):
             np.testing.assert_allclose(
                 cinn_res[i], dy_res[i], rtol=1e-6, atol=1e-6
             )
-        else:
-            pass
 
 
 if __name__ == '__main__':

python/paddle/fluid/tests/unittests/dygraph_to_static/test_cinn_prim_gelu.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import platform
 import unittest
 
 import numpy as np
@@ -98,14 +97,9 @@ class TestPrimForwardAndBackward(unittest.TestCase):
         self.assertTrue('gelu' not in fwd_ops)
 
     def test_cinn_prim(self):
-        plat = platform.system()
-        if plat == "Linux":
         for shape in self.shapes:
             for dtype in self.dtypes:
-                    if (
-                        paddle.device.get_device() == "cpu"
-                        and dtype == "float16"
-                    ):
+                if paddle.device.get_device() == "cpu" and dtype == "float16":
                     print("need pass this case")
                     continue
                 data = generate_data(shape, dtype)
@@ -121,9 +115,6 @@ class TestPrimForwardAndBackward(unittest.TestCase):
                         atol=TOLERANCE[dtype]['atol'],
                     )
 
-        else:
-            pass
-
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/fluid/tests/unittests/prim/composite_ops/CMakeLists.txt

@@ -10,11 +10,6 @@ file(
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
 string(REPLACE ".py" "" TEST_OPS_GRAD "${TEST_OPS_GRAD}")
 
-if(WIN32 OR APPLE)
-  # TODO: Fix these unittests failed on Windows and MAC.
-  list(REMOVE_ITEM TEST_OPS ${TEST_OPS_GRAD})
-endif()
-
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
 endforeach()

python/paddle/incubate/autograd/composite_rules.py

@@ -34,7 +34,7 @@ def softmax_composite(x, axis):
     """define composite rule of op softmax"""
     if not x.shape:
         # do not return 1, to ensure gradients
-        res = divide(x + 1e-5, x + 1e-5)
+        res = exp(x - x)
         return res
     max_temp = max(x, axis, keepdim=True)
     max_temp.stop_gradient = True

tools/check_file_diff_approvals.sh

@@ -83,6 +83,7 @@ API_FILES=("CMakeLists.txt"
            "paddle/fluid/prim/api/composite_backward/composite_backward_api.h"
            "paddle/fluid/prim/api/manual_prim/prim_manual_api.h"
            "python/paddle/incubate/autograd/composite_rules.py"
+	   "python/paddle/incubate/autograd/primitives.py"
            )
 
 approval_line=`curl -H "Authorization: token ${GITHUB_API_TOKEN}" https://api.github.com/repos/PaddlePaddle/Paddle/pulls/${GIT_PR_ID}/reviews?per_page=10000`
@@ -210,7 +211,7 @@ for API_FILE in ${API_FILES[*]}; do
             check_approval 1 JiabinYang cxxly xiaoguoguo626807
       elif [ "${API_FILE}" == "python/paddle/incubate/autograd/primitives.py" ] || [ "${API_FILE}" == "python/paddle/incubate/autograd/composite_rules.py" ]; then
             echo_line="You must have one RD (cyber-pioneer(chenzhuo), JiabinYang) approval for changing ${API_FILE} , which manages the composite rules.\n"
-            check_approval cyber-pioneer JiabinYang
+            check_approval 1 cyber-pioneer JiabinYang
 
       else
           echo_line="You must have one RD (XiaoguangHu01,chenwhql,zhiqiu,Xreki,luotao1,qili93) approval for ${API_FILE}, which manages the underlying code for fluid.\n"