未验证 提交 2c500c30 编写于 作者: H Huihuang Zheng 提交者: GitHub

Handle Windows flaky test (#25070)

As the title
上级 5822862d
...@@ -466,6 +466,12 @@ class TestCondBackward(unittest.TestCase): ...@@ -466,6 +466,12 @@ class TestCondBackward(unittest.TestCase):
lambda: batchnorm_fc_with_inputs(img, label, class_num=10)) lambda: batchnorm_fc_with_inputs(img, label, class_num=10))
for use_parallel_exe in [False, True]: for use_parallel_exe in [False, True]:
if use_parallel_exe and os.name == "nt":
print(
"Skip use_parallel_exe=True in Windows because of flaky test when using PE under old Windows machine"
)
continue
self.backward_value_helper(cond_func, self.backward_value_helper(cond_func,
core.is_compiled_with_cuda(), core.is_compiled_with_cuda(),
use_parallel_exe) use_parallel_exe)
...@@ -487,6 +493,12 @@ class TestCondBackward(unittest.TestCase): ...@@ -487,6 +493,12 @@ class TestCondBackward(unittest.TestCase):
lambda: branch(i, img, label)) lambda: branch(i, img, label))
for use_parallel_exe in [False, True]: for use_parallel_exe in [False, True]:
if use_parallel_exe and os.name == "nt":
print(
"Skip use_parallel_exe=True in Windows because of flaky test when using PE under old Windows machine"
)
continue
self.backward_value_helper(cond_func_simple_net_at_true, self.backward_value_helper(cond_func_simple_net_at_true,
core.is_compiled_with_cuda(), core.is_compiled_with_cuda(),
use_parallel_exe) use_parallel_exe)
...@@ -514,6 +526,11 @@ class TestCondBackward(unittest.TestCase): ...@@ -514,6 +526,11 @@ class TestCondBackward(unittest.TestCase):
lambda: branch(i, img, label, False)) lambda: branch(i, img, label, False))
for use_parallel_exe in [False, True]: for use_parallel_exe in [False, True]:
if use_parallel_exe and os.name == "nt":
print(
"Skip use_parallel_exe=True in Windows because of flaky test when using PE under old Windows machine"
)
continue
self.backward_value_helper(cond_func, self.backward_value_helper(cond_func,
core.is_compiled_with_cuda(), core.is_compiled_with_cuda(),
use_parallel_exe) use_parallel_exe)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册