Allow lowering to look at sound operations

Author: pavpanchekha

src/core/egg-herbie.rkt

@@ -539,6 +539,10 @@
                                 (rule (sym-append 'lower-unsound- impl)
                                       (add-unsound spec-expr)
                                       impl-expr
+                                      '(lowering))
+                                (rule (sym-append 'lower-sound- impl)
+                                      (add-sound-with-wildcard spec-expr)
+                                      impl-expr
                                       '(lowering))))))))
 
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

src/core/rules.rkt

@@ -8,7 +8,8 @@
 (provide *rules*
          *sound-rules*
          (struct-out rule)
-         add-unsound)
+         add-unsound
+         add-sound-with-wildcard)
 
 ;; A rule represents "find-and-replacing" `input` by `output`. Both
 ;; are patterns, meaning that symbols represent pattern variables.
@@ -36,6 +37,13 @@
     [(list op args ...) (cons (sym-append "unsound-" op) (map add-unsound args))]
     [_ expr]))
 
+(define (add-sound-with-wildcard expr)
+  (match expr
+    [(list op args ...)
+     (cons (sym-append "sound-" op)
+           (append (map add-sound-with-wildcard args) (list '_)))]
+    [_ expr]))
+
 (define-syntax define-rule
   (syntax-rules ()
     [(define-rule rname group input output)