Skip to content

Commit be632d8

Browse files
authored
open the infer_symbol_check in more cinn tests (#64690)
1 parent 842e652 commit be632d8

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

test/ir/pir/cinn/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ if(WITH_GPU)
1818
COMMAND
1919
${CMAKE_COMMAND} -E env
2020
PYTHONPATH=${CMAKE_BINARY_DIR}:${CMAKE_BINARY_DIR}/python/:$ENV{PYTHONPATH}
21-
FLAGS_enable_pir_api=1 FLAGS_prim_all=True
21+
FLAGS_enable_pir_api=1 FLAGS_prim_all=True FLAGS_check_infer_symbolic=1
2222
FLAGS_cinn_new_group_scheduler=1 FLAGS_cinn_bucket_compile=1
2323
FLAGS_group_schedule_tiling_first=1 ${PYTHON_EXECUTABLE}
2424
${CMAKE_CURRENT_SOURCE_DIR}/${cinn_pir_test_name}.py

test/ir/pir/cinn/inference/CMakeLists.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,9 @@ if(WITH_GPU)
1212
${CMAKE_COMMAND} -E env
1313
PYTHONPATH=${CMAKE_BINARY_DIR}:${CMAKE_BINARY_DIR}/python/:$ENV{PYTHONPATH}
1414
FLAGS_prim_enable_dynamic=True FLAGS_prim_all=True
15-
FLAGS_enable_pir_api=1 FLAGS_cinn_bucket_compile=True
16-
FLAGS_group_schedule_tiling_first=1 ${PYTHON_EXECUTABLE}
15+
FLAGS_check_infer_symbolic=1 FLAGS_enable_pir_api=1
16+
FLAGS_cinn_bucket_compile=True FLAGS_group_schedule_tiling_first=1
17+
${PYTHON_EXECUTABLE}
1718
${CMAKE_CURRENT_SOURCE_DIR}/${cinn_pir_test_name}.py
1819
WORKING_DIRECTORY ${CMAKE_BINARY_DIR})
1920
set_tests_properties(${cinn_pir_test_name} PROPERTIES LABELS

test/ir/pir/cinn/sub_graphs/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,9 @@ if(WITH_GPU)
9595
${CMAKE_COMMAND} -E env
9696
PYTHONPATH=${CMAKE_BINARY_DIR}:${CMAKE_BINARY_DIR}/python/:$ENV{PYTHONPATH}
9797
FLAGS_prim_enable_dynamic=1 FLAGS_cinn_new_group_scheduler=1
98-
FLAGS_enable_pir_api=1 FLAGS_cinn_bucket_compile=1
99-
FLAGS_group_schedule_tiling_first=1 FLAGS_cudnn_deterministic=true
100-
${PYTHON_EXECUTABLE}
98+
FLAGS_check_infer_symbolic=1 FLAGS_enable_pir_api=1
99+
FLAGS_cinn_bucket_compile=1 FLAGS_group_schedule_tiling_first=1
100+
FLAGS_cudnn_deterministic=true ${PYTHON_EXECUTABLE}
101101
${CMAKE_CURRENT_SOURCE_DIR}/${cinn_sub_graph_test_name}.py
102102
WORKING_DIRECTORY ${CMAKE_BINARY_DIR})
103103
set_tests_properties(${cinn_sub_graph_test_name} PROPERTIES LABELS

0 commit comments

Comments
 (0)