diff --git a/test/prim/model/test_bert_prim_cinn.py b/test/prim/model/test_bert_prim_cinn.py index dc1be501ce30e713d9b66426bec162cb5732d2fa..39af753e1a9071e7d8bd5cc18d48ea5cfac44893 100644 --- a/test/prim/model/test_bert_prim_cinn.py +++ b/test/prim/model/test_bert_prim_cinn.py @@ -178,7 +178,6 @@ class TestBert(unittest.TestCase): paddle.set_flags( {'FLAGS_deny_cinn_ops': "gaussian_random;uniform_random"} ) - core._add_skip_comp_ops("layer_norm") dy2st_prim_cinn = train( to_static=True, enable_prim=True, enable_cinn=True )