[tvm] branch main updated: [FIX, AUTOTVM] Fix printing of measure results (#10647)

[an...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

andrewzhaoluo pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 93969db  [FIX,AUTOTVM] Fix printing of measure results (#10647)
93969db is described below

commit 93969dbebc1b528f890e4ab23c7cef8e10a1e620
Author: Tristan Konolige <tk...@octoml.ai>
AuthorDate: Wed Mar 23 21:23:08 2022 -0700

    [FIX,AUTOTVM] Fix printing of measure results (#10647)
    
    * [FIX,AUTOTVM] Fix printing of measure results
    
    The check for if the error_no was valid was wrong.
    
    * switch logic
---
 python/tvm/autotvm/measure/measure.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/tvm/autotvm/measure/measure.py b/python/tvm/autotvm/measure/measure.py
index ea7de35..c9b82cd 100644
--- a/python/tvm/autotvm/measure/measure.py
+++ b/python/tvm/autotvm/measure/measure.py
@@ -55,9 +55,9 @@ class MeasureResult(namedtuple("MeasureResult", ["costs", "error_no", "all_cost"
 
     def __repr__(self):
         error_no_str = (
-            str(self.error_no)
-            if self.error_no not in MeasureErrorNo
-            else str(MeasureErrorNo(self.error_no))
+            str(MeasureErrorNo(self.error_no))
+            if isinstance(self.error_no, (MeasureErrorNo, int))
+            else str(self.error_no)
         )
         return (
             f"{self.__class__.__name__}(costs={self.costs!r}, error_no={error_no_str}, "