File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -311,6 +311,8 @@ function(cc_test TARGET_NAME)
311
311
set_property (TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_cpu_deterministic=true )
312
312
set_property (TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_init_allocated_mem=true )
313
313
set_property (TEST ${TARGET_NAME} PROPERTY ENVIRONMENT FLAGS_cudnn_deterministic=true )
314
+ # No unit test should exceed 10 minutes.
315
+ set_tests_properties (${TARGET_NAME} PROPERTIES TIMEOUT 600 )
314
316
endif ()
315
317
endfunction (cc_test )
316
318
@@ -629,6 +631,8 @@ function(py_test TARGET_NAME)
629
631
PYTHONPATH=${PADDLE_BINARY_DIR}/python ${py_test_ENVS}
630
632
${PYTHON_EXECUTABLE} -u ${py_test_SRCS} ${py_test_ARGS}
631
633
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} )
634
+ # No unit test should exceed 10 minutes.
635
+ set_tests_properties (${TARGET_NAME} PROPERTIES TIMEOUT 600 )
632
636
endif ()
633
637
endfunction ()
634
638
Original file line number Diff line number Diff line change @@ -81,10 +81,6 @@ class CompileTimeInferShapeContext : public InferShapeContext {
81
81
" The %s[%d] is @EMPTY@" , out, j);
82
82
auto *in_var = block_.FindVarRecursive (Inputs (in)[i]);
83
83
auto *out_var = block_.FindVarRecursive (Outputs (out)[j]);
84
- if (in_var->GetType () != proto::VarType::LOD_TENSOR) {
85
- VLOG (3 ) << " input " << in << " is not LodTensor" ;
86
- return ;
87
- }
88
84
PADDLE_ENFORCE_EQ (in_var->GetType (), proto::VarType::LOD_TENSOR,
89
85
" The %d-th output of Output(%s) must be LoDTensor." , j,
90
86
out);
You can’t perform that action at this time.
0 commit comments