From 47625fc93a20b7286e704ffe5414ccd7c17497bc Mon Sep 17 00:00:00 2001 From: xadupre Date: Thu, 20 Jul 2023 17:05:49 +0000 Subject: [PATCH] remove debug --- onnxruntime/test/python/quantization/op_test_utils.py | 2 +- onnxruntime/test/python/quantization/test_op_gemm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/onnxruntime/test/python/quantization/op_test_utils.py b/onnxruntime/test/python/quantization/op_test_utils.py index 960cf0b5042c..b13521ef1b45 100644 --- a/onnxruntime/test/python/quantization/op_test_utils.py +++ b/onnxruntime/test/python/quantization/op_test_utils.py @@ -187,7 +187,7 @@ def check_model_correctness( origin_sess = onnxruntime.InferenceSession(model_path_origin, sess_options=sess_options, providers=providers) origin_results = origin_sess.run([], inputs) - if True or pv.Version(onnx.__version__) >= pv.Version("1.16.0"): + if pv.Version(onnx.__version__) >= pv.Version("1.16.0"): ref = ReferenceEvaluator(model_path_to_check, new_ops=[QGemm]) target_results = ref.run(None, inputs) testcase.assertEqual(len(origin_results), len(target_results), "result count are different") diff --git a/onnxruntime/test/python/quantization/test_op_gemm.py b/onnxruntime/test/python/quantization/test_op_gemm.py index 5fc5d062bc8c..5c02e3abf816 100644 --- a/onnxruntime/test/python/quantization/test_op_gemm.py +++ b/onnxruntime/test/python/quantization/test_op_gemm.py @@ -734,7 +734,7 @@ def test_qgemm_ref_uint8_specific_example(self): if __name__ == "__main__": - # TestOpGemm().test_quantize_gemm_e4m3fn_same() + TestOpGemm().test_quantize_gemm_e4m3fn_same() # TestOpGemm().test_qgemm_ref_uint8_specific_example() # TestOpGemm()._test_dummy() # stop