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

Clean up unboxed function parsing loop.

Browse Source 
original commit: d882abe47bfdd899bdb5371cc6f604bdff97e13e
This commit is contained in:
Eric Dobson 2013-09-18 21:00:36 -07:00
parent c49290c3f3
commit 5deef7e6b6
1 changed files with 20 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
pkgs/typed-racket-pkgs/typed-racket-lib/typed-racket/optimizer/unboxed-let.rkt
View File

@ -86,32 +86,26 @@
(syntax-parse (cadr p)
#:literal-sets (kernel-literals)
[(#%plain-lambda params body ...)
;; keep track of the param # of each param that can be
;; unboxed
(let loop ((unboxed '())
(boxed '())
(i 0)
(params (syntax->list #'params))
(doms doms))
(cond [(null? params)
;; done. can we unbox anything?
(and (> (length unboxed) 0)
;; if so, add to the table of functions with
;; unboxed params, so we can modify its call
;; sites, its body and its header
(add-unboxed-fun! fun-name unboxed boxed))]
[(and (equal? (car doms) -FloatComplex)
(could-be-unboxed-in?
(car params) #'(begin body ...)))
;; we can unbox
(log-optimization "unboxed var -> table"
arity-raising-opt-msg
(car params))
(loop (cons i unboxed) boxed
(add1 i) (cdr params) (cdr doms))]
[else ; can't unbox
(loop unboxed (cons i boxed)
(add1 i) (cdr params) (cdr doms))]))]
(define-values (unboxed boxed)
(for/fold ([unboxed empty] [boxed empty])
([param (in-syntax #'params)]
[dom doms]
[i (in-naturals)])
(cond
[(and (equal? dom -FloatComplex)
(could-be-unboxed-in?
param
#'(begin body ...)))
;; we can unbox
(log-optimization "unboxed var -> table" arity-raising-opt-msg param)
(values (cons i unboxed) boxed)]
[else (values unboxed (cons i boxed))])))
;; can we unbox anything?
(and (> (length unboxed) 0)
;; if so, add to the table of functions with
;; unboxed params, so we can modify its call
;; sites, its body and its header
(add-unboxed-fun! fun-name unboxed boxed))]
[_ #f])]
[_ #f])))))
rest)))