Skip to content

Commit 909a5d1

Browse files
committed
correctly outputing expressions seperating them by eclasses
1 parent 6013b52 commit 909a5d1

File tree

1 file changed

+8
-19
lines changed

1 file changed

+8
-19
lines changed

src/core/egg-herbie.rkt

Lines changed: 8 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1194,29 +1194,18 @@
11941194
(match cmd
11951195
[`(single . ,extractor) ; single expression extraction
11961196
(define regraph (make-regraph egg-graph))
1197-
(define nodenumber 0)
1198-
(displayln "%%%%%%%%%%%%%%%%%%%%%%%%%%%%")
1199-
(define eclasses (regraph-eclasses regraph))
1200-
(define n (vector-length eclasses))
1201-
(for ([id (in-range n)])
1202-
(displayln "___________________")
1203-
1204-
(define eclass (vector-ref eclasses id))
1205-
(for ([enode eclass])
1206-
(display nodenumber)
1207-
(display ": ")
1208-
(displayln enode)
1209-
(set! nodenumber (+ nodenumber 1))
1210-
;(displayln "___________________")
1211-
)
1212-
)
1213-
1214-
(displayln "%%%%%%%%%%%%%%%%%%%%%%%%%%%%")
1197+
1198+
(displayln "%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%")
12151199
(define extract-id (extractor regraph))
12161200
(define reprs (egg-runner-reprs runner))
12171201
(for/list ([id (in-list root-ids)]
12181202
[repr (in-list reprs)])
1219-
(regraph-extract-best regraph extract-id id repr))]
1203+
(define eclass (regraph-extract-variants regraph extract-id id repr))
1204+
(for([expr eclass])
1205+
(displayln expr))
1206+
(regraph-extract-best regraph extract-id id repr)
1207+
)
1208+
]
12201209
[`(multi . ,extractor) ; multi expression extraction
12211210
(define regraph (make-regraph egg-graph))
12221211
(define extract-id (extractor regraph))

0 commit comments

Comments
 (0)