remove debug

xadupre · Jul 20, 2023 · 47625fc · 47625fc
1 parent 8bbe353
commit 47625fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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