@@ -1382,13 +1382,10 @@ def try_rewrite(
1382
1382
if not check_match_result :
1383
1383
# If check function was provided, but it failed, return the reason for failure to the tracer.
1384
1384
if isinstance (check_match_result , MatchResult ):
1385
- if check_match_result ._failure_nodes_and_values :
1386
- match .fail (
1387
- check_match_result .reason ,
1388
- check_match_result ._failure_nodes_and_values ,
1389
- )
1390
- else :
1391
- match .fail (check_match_result .reason )
1385
+ match .fail (
1386
+ check_match_result .reason ,
1387
+ check_match_result ._failure_nodes_and_values ,
1388
+ )
1392
1389
if tracer :
1393
1390
tracer .log (
1394
1391
self , graph_or_function , node , match , MatchStatus .CONDITION_FAILED
@@ -1827,18 +1824,9 @@ def print(self):
1827
1824
print ("Graph matching failed." )
1828
1825
failure_nodes_and_values = self .match_result ._failure_nodes_and_values
1829
1826
print ("Failure at or around nodes/values:" )
1830
-
1831
1827
if failure_nodes_and_values :
1832
- if isinstance (failure_nodes_and_values , list ):
1833
- for failure_cause in failure_nodes_and_values :
1834
- if isinstance (failure_cause , ir .Node ):
1835
- failure_cause .display ()
1836
- elif isinstance (failure_cause , ir .Value ):
1837
- print (failure_cause )
1838
- elif isinstance (failure_nodes_and_values , ir .Node ):
1839
- failure_nodes_and_values .display ()
1840
- elif isinstance (failure_nodes_and_values , ir .Value ):
1841
- print (failure_nodes_and_values )
1828
+ for failure_cause in failure_nodes_and_values :
1829
+ failure_cause .display ()
1842
1830
print ("Matched nodes:" )
1843
1831
import onnxscript .rewriter ._ir_utils as ir_utils
1844
1832
0 commit comments