From a677a1e2cdeb7f0fdc05a00f51aea289451c6fc6 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Fri, 29 Sep 2023 05:41:29 -0700 Subject: [PATCH] winograd test prints op count --- test/test_winograd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/test_winograd.py b/test/test_winograd.py index ab956fe0d9..b062946a1f 100644 --- a/test/test_winograd.py +++ b/test/test_winograd.py @@ -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()