未验证 提交 d913bc98 编写于 作者: A Allen Guo 提交者: GitHub

fix ut timeout 2 (#45233) (#46867)

上级 bd89be12
...@@ -11,10 +11,14 @@ if(WITH_IPU) ...@@ -11,10 +11,14 @@ if(WITH_IPU)
endforeach() endforeach()
set_tests_properties(test_conv_op_ipu PROPERTIES TIMEOUT 300) set_tests_properties(test_conv_op_ipu PROPERTIES TIMEOUT 300)
set_tests_properties(test_elemetwise_x_op_ipu PROPERTIES TIMEOUT 300) set_tests_properties(test_elemetwise_x_op_ipu PROPERTIES TIMEOUT 600)
set_tests_properties(test_reduce_x_op_ipu PROPERTIES TIMEOUT 600) set_tests_properties(test_reduce_x_op_ipu PROPERTIES TIMEOUT 800)
set_tests_properties(test_save_load_ipu PROPERTIES TIMEOUT 600) set_tests_properties(test_save_load_ipu PROPERTIES TIMEOUT 600)
set_tests_properties(test_activation_ops_ipu PROPERTIES TIMEOUT 600) set_tests_properties(test_activation_ops_ipu PROPERTIES TIMEOUT 600)
set_tests_properties(test_unary_ops_ipu PROPERTIES TIMEOUT 600)
set_tests_properties(test_greater_op_ipu PROPERTIES TIMEOUT 400)
set_tests_properties(test_conv2d_transpose_op_ipu PROPERTIES TIMEOUT 300)
set_tests_properties(test_model_parallel_ipu PROPERTIES TIMEOUT 300)
add_subdirectory(custom_ops) add_subdirectory(custom_ops)
......
...@@ -204,6 +204,14 @@ class IPUOpTest(IPUTest): ...@@ -204,6 +204,14 @@ class IPUOpTest(IPUTest):
if self.is_fp16_mode(exec_mode): if self.is_fp16_mode(exec_mode):
ipu_strategy.set_precision_config(enable_fp16=True) ipu_strategy.set_precision_config(enable_fp16=True)
IPUOpTest.cast_model_to_fp16(self.main_prog) IPUOpTest.cast_model_to_fp16(self.main_prog)
# TODO(ipu) remove in the future version of popart
# keep the log clean, no side effects for tests without profiling
ipu_strategy.set_options(
{'engine_options': {
'debug.retainDebugInformation': 'false'
}})
program = paddle.static.IpuCompiledProgram( program = paddle.static.IpuCompiledProgram(
self.main_prog, self.main_prog,
ipu_strategy=ipu_strategy).compile(self.feed_list, ipu_strategy=ipu_strategy).compile(self.feed_list,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册