winograd test prints op count

2023-09-29 05:41:29 -07:00
parent 4ff35e2b97
commit a677a1e2cd
1 changed files with 2 additions and 1 deletions
@@ -22,7 +22,8 @@ class TestWinograd(unittest.TestCase):

    for i,s in enumerate(sched):
      if s[0].op in LoadOps: continue
-      with Timing(f"linearize {i}: "):
+      ops = s[0].get_lazyops()
+      with Timing(f"linearize {i} with {len(ops):4d} ops: "):
        l = Linearizer(s[0])
        l.hand_coded_optimizations()
        l.linearize()