[NewIR] No.10 Migrate silu into pir

python/paddle/nn/functional/activation.py

@@ -14,7 +14,7 @@
 
 import paddle
 from paddle import _C_ops, _legacy_C_ops, in_dynamic_mode
-from paddle.framework import core
+from paddle.framework import core, in_dynamic_or_new_ir_mode
 from paddle.utils.inplace_utils import inplace_apis_in_dygraph_only
 
 from ...base.data_feeder import check_dtype, check_variable_and_dtype
@@ -1053,14 +1053,13 @@ def silu(x, name=None):
             [0.73105860, 1.76159406, 2.85772228, 3.92805505])
     """
 
-    if in_dynamic_mode():
+    if in_dynamic_or_new_ir_mode():
         return _C_ops.silu(x)
     else:
         check_variable_and_dtype(
             x,
             'x',
             [
-                'float16',
                 'uint16',
                 'float32',
                 'float64',

test/legacy_test/test_activation_op.py

@@ -424,6 +424,14 @@ def init_dtype(self):
         self.dtype = np.complex128
 
 
+class TestSilu_NewIR(TestSilu):
+    def test_check_output(self):
+        self.check_output(check_new_ir=True)
+
+    def test_checkout_grad(self):
+        self.check_grad(['X'], 'Out', check_prim=True, check_new_ir=True)
+
+
 class TestSiluAPI(unittest.TestCase):
     # test paddle.nn.Silu, paddle.nn.functional.silu
     def setUp(self):