Skip to content

Commit

Permalink
[PIR+CINN]fix timeout from test_llama_inference/forward (#63680)
Browse files Browse the repository at this point in the history
* [PIR+CINN]fix timeout from test_llama_inference/forward

* test=document_fix
  • Loading branch information
Aurelius84 authored Apr 19, 2024
1 parent d8a230e commit 69d3115
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions test/ir/pir/cinn/inference/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -19,8 +19,9 @@ if(WITH_GPU)
set_tests_properties(${cinn_pir_test_name} PROPERTIES LABELS
"RUN_TYPE=CINN")
endforeach()
set_tests_properties(test_llama_inference PROPERTIES TIMEOUT 120)
set_tests_properties(test_llama_forward PROPERTIES TIMEOUT 120)

set_tests_properties(test_llama_inference PROPERTIES TIMEOUT 300)
set_tests_properties(test_llama_forward PROPERTIES TIMEOUT 300)

add_test(
NAME test_llama_postprocess_cinn
Expand Down

0 comments on commit 69d3115

Please sign in to comment.