fix test_layernorm_shift_partition_pass time out (#51612)

b642461d · wenbin · GitHub · 46d6080d · b642461d
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt .../paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt +2 -0

未找到文件。
--- a/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt
+++ b/python/paddle/fluid/tests/unittests/ir/inference/CMakeLists.txt
@@ -251,6 +251,8 @@ if(WITH_GPU AND TENSORRT_FOUND)
      set_tests_properties(test_map_matmul_v2_to_mul_pass PROPERTIES TIMEOUT
                                                                     360)
      set_tests_properties(test_map_matmul_to_mul_pass PROPERTIES TIMEOUT 360)
+      set_tests_properties(test_layernorm_shift_partition_pass
+                           PROPERTIES TIMEOUT 360)
    else()
      set_tests_properties(test_matmul_scale_fuse_pass PROPERTIES TIMEOUT 60)
      set_tests_properties(test_matmul_v2_scale_fuse_pass PROPERTIES TIMEOUT 60)