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 @@ -85,10 +85,6 @@ class CompileTimeInferShapeContext : public InferShapeContext {
85
85
VLOG (3 ) << " input " << in << " is not LodTensor" ;
86
86
return ;
87
87
}
88
- PADDLE_ENFORCE_EQ (in_var->GetType (), proto::VarType::LOD_TENSOR,
89
- " The %d-th output of Output(%s) must be LoDTensor." , j,
90
- out);
91
-
92
88
out_var->SetLoDLevel (in_var->GetLoDLevel ());
93
89
}
94
90
You can’t perform that action at this time.
0 commit comments