saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

nGraph EP disable new quant tests (#920)

Browse source 
* Disable quant op test cases

* enable ngraph quant ops
This commit is contained in:
nivas-x86 2019-04-26 14:21:20 -07:00 committed by jywu-msft
parent 3812c881f7
commit b5ea3973c4
3 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/ngraph/ngraph_execution_provider.cc
View file

@ -288,7 +288,7 @@ static std::map<std::string, std::set<std::string>> GetNgSupportedOps(const int
std::map<std::string, std::set<std::string>> ng_supported_ops;
ng_supported_ops.emplace(kOnnxDomain, ngraph::onnx_import::get_supported_operators(onnx_opset, kOnnxDomain));
const std::set<std::string> ng_disabled_ops = {"DequantizeLinear", "QLinearConv", "QuantizeLinear"}; //Place-holder for ops not supported.
const std::set<std::string> ng_disabled_ops = {}; //Place-holder for ops not supported.
for (const auto& disabled_op : ng_disabled_ops) {
ng_supported_ops.at(kOnnxDomain).erase(disabled_op);

6
onnxruntime/test/onnx/main.cc
View file

@ -359,6 +359,12 @@ int real_main(int argc, char* argv[], OrtEnv** p_env) {
{"mod_float_mixed_sign_example", "faulty test"}
};
#ifdef USE_NGRAPH
broken_tests["dequantizelinear"] = "ambiguity in scalar dimensions [] vs [1]";
broken_tests["qlinearconv"] = "ambiguity in scalar dimensions [] vs [1]";
broken_tests["quantizelinear"] = "ambiguity in scalar dimensions [] vs [1]";
#endif
#ifdef USE_CUDA
broken_tests["mxnet_arcface"] = "result mismatch";
#endif

3
onnxruntime/test/python/onnx_backend_test_series.py
View file

@ -44,6 +44,7 @@ backend_test.exclude(r'('
'|^test_convtranspose_1d_cpu.*'
'|^test_convtranspose_3d_cpu.*'
'|^test_constantofshape_*.*'
'|^test_dequantizelinear_cpu.*'
'|^test_AvgPool1d_cpu.*'
'|^test_AvgPool1d_stride_cpu.*'
@ -84,6 +85,8 @@ backend_test.exclude(r'('
'|^test_zfnet512_cpu.*'
'|^test_gru_seq_length_cpu.*'
'|^test_mvn_cpu.*'
'|^test_qlinearconv_cpu.*'
'|^test_quantizelinear_cpu.*'
'|^test_mod_float_mixed_sign_example.*'
'|^test_reversesequence_batch_cpu.*'
'|^test_reversesequence_time_cpu.*'