未验证 提交 3de2206c 编写于 作者: W WangZhen 提交者: GitHub

Remove cinn deny ops for bert test (#52897)

上级 cf3ddf24
...@@ -57,19 +57,21 @@ DY2ST_CINN_GT = [ ...@@ -57,19 +57,21 @@ DY2ST_CINN_GT = [
10.112163543701172, 10.112163543701172,
9.969393730163574, 9.969393730163574,
] ]
DY2ST_PRIM_CINN_GT = [ DY2ST_PRIM_CINN_GT = [
11.144556999206543, 10.976988792419434,
10.343620300292969, 10.345282554626465,
10.330279350280762, 10.33430004119873,
10.276118278503418, 10.275235176086426,
10.222086906433105, 10.22808837890625,
10.194628715515137, 10.200264930725098,
10.149020195007324, 10.161019325256348,
10.096250534057617, 10.10872745513916,
10.104615211486816, 10.121539115905762,
9.985644340515137, 9.990568161010742,
] ]
if core.is_compiled_with_cuda(): if core.is_compiled_with_cuda():
paddle.set_flags({'FLAGS_cudnn_deterministic': True}) paddle.set_flags({'FLAGS_cudnn_deterministic': True})
...@@ -175,9 +177,6 @@ class TestBert(unittest.TestCase): ...@@ -175,9 +177,6 @@ class TestBert(unittest.TestCase):
"paddle is not compiled with CINN and CUDA", "paddle is not compiled with CINN and CUDA",
) )
def test_prim_cinn(self): def test_prim_cinn(self):
paddle.set_flags(
{'FLAGS_deny_cinn_ops': "gaussian_random;uniform_random"}
)
dy2st_prim_cinn = train( dy2st_prim_cinn = train(
to_static=True, enable_prim=True, enable_cinn=True to_static=True, enable_prim=True, enable_cinn=True
) )
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册