File tree 3 files changed +7
-6
lines changed
3 files changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -18,7 +18,7 @@ if(WITH_GPU)
18
18
COMMAND
19
19
${CMAKE_COMMAND} -E env
20
20
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
22
22
FLAGS_cinn_new_group_scheduler=1 FLAGS_cinn_bucket_compile=1
23
23
FLAGS_group_schedule_tiling_first=1 ${PYTHON_EXECUTABLE}
24
24
${CMAKE_CURRENT_SOURCE_DIR} /${cinn_pir_test_name} .py
Original file line number Diff line number Diff line change @@ -12,8 +12,9 @@ if(WITH_GPU)
12
12
${CMAKE_COMMAND} -E env
13
13
PYTHONPATH=${CMAKE_BINARY_DIR} :${CMAKE_BINARY_DIR} /python/:$ENV{PYTHONPATH}
14
14
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}
17
18
${CMAKE_CURRENT_SOURCE_DIR} /${cinn_pir_test_name} .py
18
19
WORKING_DIRECTORY ${CMAKE_BINARY_DIR} )
19
20
set_tests_properties (${cinn_pir_test_name} PROPERTIES LABELS
Original file line number Diff line number Diff line change @@ -95,9 +95,9 @@ if(WITH_GPU)
95
95
${CMAKE_COMMAND} -E env
96
96
PYTHONPATH=${CMAKE_BINARY_DIR} :${CMAKE_BINARY_DIR} /python/:$ENV{PYTHONPATH}
97
97
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}
101
101
${CMAKE_CURRENT_SOURCE_DIR} /${cinn_sub_graph_test_name} .py
102
102
WORKING_DIRECTORY ${CMAKE_BINARY_DIR} )
103
103
set_tests_properties (${cinn_sub_graph_test_name} PROPERTIES LABELS
You can’t perform that action at this time.
0 commit comments