未验证 提交 d4699bd6 编写于 作者: R Ruibiao Chen 提交者: GitHub

Enable test_tensordot (#42932)

* Enable test_tesnsordot

* Fix CI errors

* Disable windows inference test

* Fix typos

* Fix typos

* Fix CI errors

* Remove disable_wingpu11_test
上级 01b3ccae
...@@ -205,8 +205,6 @@ endif() ...@@ -205,8 +205,6 @@ endif()
# Temporally disable test_deprecated_decorator # Temporally disable test_deprecated_decorator
list(REMOVE_ITEM TEST_OPS test_deprecated_decorator) list(REMOVE_ITEM TEST_OPS test_deprecated_decorator)
list(REMOVE_ITEM TEST_OPS test_tensordot)
if(WIN32) if(WIN32)
list(REMOVE_ITEM TEST_OPS test_multiprocess_reader_exception) list(REMOVE_ITEM TEST_OPS test_multiprocess_reader_exception)
list(REMOVE_ITEM TEST_OPS test_trainer_desc) list(REMOVE_ITEM TEST_OPS test_trainer_desc)
...@@ -1419,7 +1417,7 @@ set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120) ...@@ -1419,7 +1417,7 @@ set_tests_properties(test_index_select_op PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_ssa_graph_inference_feed_partial_data set_tests_properties(test_parallel_ssa_graph_inference_feed_partial_data
PROPERTIES TIMEOUT 120) PROPERTIES TIMEOUT 120)
set_tests_properties(test_parallel_executor_crf PROPERTIES TIMEOUT 120) set_tests_properties(test_parallel_executor_crf PROPERTIES TIMEOUT 120)
#set_tests_properties(test_tensordot PROPERTIES TIMEOUT 200) set_tests_properties(test_tensordot PROPERTIES TIMEOUT 200)
set_tests_properties(test_imperative_save_load PROPERTIES TIMEOUT 120) set_tests_properties(test_imperative_save_load PROPERTIES TIMEOUT 120)
set_tests_properties(test_partial_eager_deletion_transformer PROPERTIES TIMEOUT set_tests_properties(test_partial_eager_deletion_transformer PROPERTIES TIMEOUT
120) 120)
......
...@@ -185,35 +185,35 @@ class TestTensordotAPIFloat64(TestTensordotAPI): ...@@ -185,35 +185,35 @@ class TestTensordotAPIFloat64(TestTensordotAPI):
self.dtype = np.float64 self.dtype = np.float64
class TestTensordotAPIBroadcastCase1(TestTensordotAPI): class TestTensordotAPIBroadcastCase1(TestTensordotAPIFloat64):
def set_input_shape(self): def set_input_shape(self):
self.x_shape = [1, 1, 1, 5] self.x_shape = [1, 1, 1, 5]
self.y_shape = [1, 5, 1, 1] self.y_shape = [1, 5, 1, 1]
class TestTensordotAPIBroadcastCase2(TestTensordotAPI): class TestTensordotAPIBroadcastCase2(TestTensordotAPIFloat64):
def set_input_shape(self): def set_input_shape(self):
self.x_shape = [1, 5, 5, 5] self.x_shape = [1, 5, 5, 5]
self.y_shape = [1, 1, 1, 5] self.y_shape = [1, 1, 1, 5]
class TestTensordotAPIBroadcastCase3(TestTensordotAPI): class TestTensordotAPIBroadcastCase3(TestTensordotAPIFloat64):
def set_input_shape(self): def set_input_shape(self):
self.x_shape = [5, 5, 5, 1] self.x_shape = [5, 5, 5, 1]
self.y_shape = [5, 5, 1, 5] self.y_shape = [5, 5, 1, 5]
class TestTensordotAPIBroadcastCase4(TestTensordotAPI): class TestTensordotAPIBroadcastCase4(TestTensordotAPIFloat64):
def set_input_shape(self): def set_input_shape(self):
self.x_shape = [5, 5, 5, 1] self.x_shape = [5, 5, 5, 1]
self.y_shape = [1, 1, 1, 1] self.y_shape = [1, 1, 1, 1]
class TestTensordotAPIBroadcastCase5(TestTensordotAPI): class TestTensordotAPIBroadcastCase5(TestTensordotAPIFloat64):
def set_input_shape(self): def set_input_shape(self):
self.x_shape = [1, 1, 5, 5] self.x_shape = [1, 1, 5, 5]
......
...@@ -72,12 +72,6 @@ disable_win_trt_test="^test_trt_convert_conv2d$|\ ...@@ -72,12 +72,6 @@ disable_win_trt_test="^test_trt_convert_conv2d$|\
^test_trt_convert_matmul$|\ ^test_trt_convert_matmul$|\
^test_trt_convert_scale$" ^test_trt_convert_scale$"
# /*=============Fixed Disabled Windows CUDA11.x MKL(PR-CI-Windows-Inference) unittests=================*/
# TODO: fix these unittest that is bound to fail
disable_wingpu11_test="^test_autograd_functional_dynamic$|\
^disable_wingpu_test$"
# /*==========Fixed Disabled Windows CUDA11.x inference_api_test(PR-CI-Windows-Inference) unittests=============*/ # /*==========Fixed Disabled Windows CUDA11.x inference_api_test(PR-CI-Windows-Inference) unittests=============*/
disable_win_inference_test="^trt_quant_int8_yolov3_r50_test$|\ disable_win_inference_test="^trt_quant_int8_yolov3_r50_test$|\
^test_trt_dynamic_shape_ernie$|\ ^test_trt_dynamic_shape_ernie$|\
...@@ -183,7 +177,9 @@ disable_win_inference_test="^trt_quant_int8_yolov3_r50_test$|\ ...@@ -183,7 +177,9 @@ disable_win_inference_test="^trt_quant_int8_yolov3_r50_test$|\
^test_unsqueeze2_eltwise_fuse_pass$|\ ^test_unsqueeze2_eltwise_fuse_pass$|\
^test_parallel_executor_seresnext_with_fuse_all_reduce_gpu$|\ ^test_parallel_executor_seresnext_with_fuse_all_reduce_gpu$|\
^test_parallel_executor_seresnext_with_reduce_gpu$|\ ^test_parallel_executor_seresnext_with_reduce_gpu$|\
^test_api_impl$" ^test_api_impl$|\
^test_tensordot$|\
^disable_wingpu_test$"
# /*==========Fixed Disabled Windows CPU OPENBLAS((PR-CI-Windows-OPENBLAS)) unittests==============================*/ # /*==========Fixed Disabled Windows CPU OPENBLAS((PR-CI-Windows-OPENBLAS)) unittests==============================*/
...@@ -281,7 +277,7 @@ bash $PADDLE_ROOT/tools/check_added_ut_win.sh ...@@ -281,7 +277,7 @@ bash $PADDLE_ROOT/tools/check_added_ut_win.sh
rm -rf $PADDLE_ROOT/tools/check_added_ut_win.sh rm -rf $PADDLE_ROOT/tools/check_added_ut_win.sh
if [ -f "$PADDLE_ROOT/added_ut" ];then if [ -f "$PADDLE_ROOT/added_ut" ];then
added_uts=^$(awk BEGIN{RS=EOF}'{gsub(/\n/,"$|^");print}' $PADDLE_ROOT/added_ut)$ added_uts=^$(awk BEGIN{RS=EOF}'{gsub(/\n/,"$|^");print}' $PADDLE_ROOT/added_ut)$
ctest -R "(${added_uts})" -E "$disable_wingpu11_test" --output-on-failure -C Release --repeat-until-fail 3;added_ut_error=$? ctest -R "(${added_uts})" -E "${disable_win_inference_test}" --output-on-failure -C Release --repeat-until-fail 3;added_ut_error=$?
rm -f $PADDLE_ROOT/added_ut rm -f $PADDLE_ROOT/added_ut
if [ "$added_ut_error" != 0 ];then if [ "$added_ut_error" != 0 ];then
echo "========================================" echo "========================================"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册