suzanne.soy/typed-racket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add missed optimizations for non complexs in complex arithmetic.

Browse Source 
original commit: 78b9a8a360007eb3bb0ccf658d6ea1e7352791e4
This commit is contained in:
Eric Dobson 2013-10-21 22:30:45 -07:00
parent ce1ef7caa7
commit 85696b0c39
2 changed files with 18 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
pkgs/typed-racket-pkgs/typed-racket-lib/typed-racket/optimizer/float-complex.rkt
View File

@ -41,6 +41,13 @@
(log-opt opt-label "Complex number unboxing."))
(define-syntax-rule (log-arity-raising-opt opt-label)
(log-opt opt-label arity-raising-opt-msg))
(define-syntax-rule (log-missed-complex-expr)
(log-missed-optimization
"Non complex value in complex arithmetic"
(string-append
"This expression has a non FloatComplex type and thus cannot "
"be promoted to unboxed arithmetic.")
this-syntax))
;; If a part is 0.0?
(define (0.0? stx)
@ -240,7 +247,8 @@
(unsafe-fl* c.imag-binding c.imag-binding))))))
(pattern (#%plain-app op:exp^ c:unboxed-float-complex-opt-expr)
#:when (subtypeof? this-syntax -FloatComplex)
#:when (or (subtypeof? this-syntax -FloatComplex)
(and (log-missed-complex-expr) #f))
#:with (real-binding imag-binding) (binding-names)
#:with scaling-factor (generate-temporary "unboxed-scaling-")
#:do [(log-unboxing-opt "unboxed unary float complex")]
@ -272,14 +280,16 @@
;; we can eliminate boxing that was introduced by the user
(pattern (#%plain-app op:make-rectangular^ real:float-arg-expr imag:float-arg-expr)
#:when (subtypeof? this-syntax -FloatComplex)
#:when (or (subtypeof? this-syntax -FloatComplex)
(and (log-missed-complex-expr) #f))
#:with (real-binding imag-binding) (binding-names)
#:do [(log-unboxing-opt "make-rectangular elimination")]
#:with (bindings ...)
#'(((real-binding) real.opt)
((imag-binding) imag.opt)))
(pattern (#%plain-app op:make-polar^ r:float-arg-expr theta:float-arg-expr)
#:when (subtypeof? this-syntax -FloatComplex)
#:when (or (subtypeof? this-syntax -FloatComplex)
(and (log-missed-complex-expr) #f))
#:with radius (generate-temporary)
#:with angle (generate-temporary)
#:with (real-binding imag-binding) (binding-names)

9
pkgs/typed-racket-pkgs/typed-racket-test/tests/typed-racket/optimizer/tests/complex-exp.rkt
View File

@ -1,9 +1,10 @@
#;#;
#<<END
TR opt: complex-exp.rkt 15:0 (- (exp 0) 1.0+2.0i) -- unboxed binary float complex
TR opt: complex-exp.rkt 15:11 1.0+2.0i -- unboxed literal
TR opt: complex-exp.rkt 15:3 (exp 0) -- float-arg-expr in complex ops
TR opt: complex-exp.rkt 15:8 0 -- float-arg-expr in complex ops
TR missed opt: complex-exp.rkt 16:3 (exp 0) -- Non complex value in complex arithmetic
TR opt: complex-exp.rkt 16:0 (- (exp 0) 1.0+2.0i) -- unboxed binary float complex
TR opt: complex-exp.rkt 16:11 1.0+2.0i -- unboxed literal
TR opt: complex-exp.rkt 16:3 (exp 0) -- float-arg-expr in complex ops
TR opt: complex-exp.rkt 16:8 0 -- float-arg-expr in complex ops
END
#<<END
0.0-2.0i