diff --git a/onnxruntime/test/providers/cpu/model_tests.cc b/onnxruntime/test/providers/cpu/model_tests.cc index 92c53989f7..7a278a4b55 100644 --- a/onnxruntime/test/providers/cpu/model_tests.cc +++ b/onnxruntime/test/providers/cpu/model_tests.cc @@ -126,6 +126,40 @@ TEST_P(ModelTest, Run) { {"momentum", "not a registered function/op", {}}, // Op not registered. {"momentum_multiple", "not a registered function/op", {}}, // Op not registered. {"nesterov_momentum", "not a registered function/op", {}}, // Op not registered. + {"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_none_no_weight", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1_mean_weight_negative_ignore_index_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_mean_weight_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index_3d", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index_4d_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_none_no_weight_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index_4d", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3_sum_weight_high_ignore_index_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_3d_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_none_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_3d", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3_none_no_weight_negative_ignore_index", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index_3d_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index_3d_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_none_weights_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_sum_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight_ignore_index", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index_3d", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3_sum_weight_high_ignore_index", "type error", {"onnx170"}}, + {"softmax_cross_entropy_sum", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3_none_no_weight_negative_ignore_index_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_none_weights", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index_4d_log_prob", "type error", {"onnx170"}}, + {"softmax_cross_entropy_none", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1_mean_weight_negative_ignore_index", "type error", {"onnx170"}}, + {"softmax_cross_entropy_input_shape_is_NCd1d2d3d4d5_mean_weight", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_weight", "type error", {"onnx170"}}, + {"softmax_cross_entropy_mean_no_weight_ignore_index_4d", "type error", {"onnx170"}}, #endif {"mask_rcnn_keras", "this model currently has an invalid contrib op version set to 10", {}}}; diff --git a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml index 28a85bbc09..223f6337b3 100644 --- a/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml +++ b/tools/ci_build/github/azure-pipelines/templates/py-packaging-stage.yml @@ -270,7 +270,7 @@ stages: timeoutInMinutes: 180 workspace: clean: all - pool: Linux-GPU-CUDA10 + pool: Onnxruntime-Linux-GPU strategy: matrix: Python36: