2007-02-10 18:51:12 -05:00
|
|
|
|
|
|
|
(module (alt-cogen)
|
|
|
|
;;; input to cogen is <Program>:
|
|
|
|
;;; <Expr> ::= (constant x)
|
|
|
|
;;; | (var)
|
|
|
|
;;; | (primref name)
|
|
|
|
;;; | (bind var* <Expr>* <Expr>)
|
|
|
|
;;; | (fix var* <FixRhs>* <Expr>)
|
|
|
|
;;; | (conditional <Expr> <Expr> <Expr>)
|
|
|
|
;;; | (seq <Expr> <Expr>)
|
|
|
|
;;; | (closure <codeloc> <var>*) ; thunk special case
|
|
|
|
;;; | (primcall op <Expr>*)
|
|
|
|
;;; | (forcall "name" <Expr>*)
|
|
|
|
;;; | (funcall <Expr> <Expr>*)
|
|
|
|
;;; | (jmpcall <label> <Expr> <Expr>*)
|
|
|
|
;;; | (appcall <Expr> <Expr>*)
|
|
|
|
;;; | (mvcall <Expr> <clambda>)
|
|
|
|
;;; <codeloc> ::= (code-loc <label>)
|
|
|
|
;;; <clambda> ::= (clambda <label> <case>* <free var>*)
|
|
|
|
;;; <case> ::= (clambda-case <info> <body>)
|
|
|
|
;;; <info> ::= (clambda-info label <arg var>* proper)
|
|
|
|
;;; <Program> ::= (codes <clambda>* <Expr>)
|
|
|
|
|
|
|
|
|
|
|
|
(define (verify-new-cogen-input x)
|
|
|
|
;;;
|
|
|
|
(define who 'verify-new-cogen-input)
|
|
|
|
;;;
|
|
|
|
(define (check-gensym x)
|
|
|
|
(unless (gensym? x)
|
|
|
|
(error who "invalid gensym ~s" x)))
|
|
|
|
;;;
|
|
|
|
(define (check-label x)
|
|
|
|
(record-case x
|
|
|
|
[(code-loc label)
|
|
|
|
(check-gensym label)]
|
|
|
|
[else (error who "invalid label ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (check-var x)
|
|
|
|
(record-case x
|
|
|
|
[(var) (void)]
|
|
|
|
[else (error who "invalid var ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (check-closure x)
|
|
|
|
(record-case x
|
|
|
|
[(closure label free*)
|
|
|
|
(check-label label)
|
|
|
|
(for-each check-var free*)]
|
|
|
|
[else (error who "invalid closure ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Expr x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) (void)]
|
|
|
|
[(var) (void)]
|
|
|
|
[(primref) (void)]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(for-each check-var lhs*)
|
|
|
|
(for-each Expr rhs*)
|
|
|
|
(Expr body)]
|
|
|
|
[(fix lhs* rhs* body)
|
|
|
|
(for-each check-var lhs*)
|
|
|
|
(for-each check-closure rhs*)
|
|
|
|
(Expr body)]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(Expr e0) (Expr e1) (Expr e2)]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(Expr e0) (Expr e1)]
|
|
|
|
[(closure) (check-closure x)]
|
|
|
|
[(primcall op arg*)
|
|
|
|
(for-each Expr arg*)]
|
|
|
|
[(forcall op arg*)
|
|
|
|
(for-each Expr arg*)]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(Expr rator)
|
|
|
|
(for-each Expr arg*)]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(check-gensym label)
|
|
|
|
(Expr rator)
|
|
|
|
(for-each Expr arg*)]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(Expr rator)
|
|
|
|
(for-each Expr arg*)]
|
|
|
|
[(mvcall rator k)
|
|
|
|
(Expr rator)
|
|
|
|
(Clambda k)]
|
|
|
|
[else (error who "invalid expr ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (check-info x)
|
|
|
|
(record-case x
|
|
|
|
[(case-info label args proper)
|
|
|
|
(check-gensym label)
|
|
|
|
(for-each check-var args)]
|
|
|
|
[else (error who "invalid case-info ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(check-info info)
|
|
|
|
(Expr body)]
|
|
|
|
[else (error who "invalid clambda-case ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(for-each check-var free*)
|
|
|
|
(for-each ClambdaCase case*)
|
|
|
|
(check-gensym label)]
|
|
|
|
[else (error who "invalid clambda ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(for-each Clambda code*)
|
|
|
|
(Expr body)]
|
|
|
|
[else (error who "invalid program ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
;;; the program so far includes both primcalls and funcalls to
|
|
|
|
;;; primrefs. This pass removes all primcalls. Once everything
|
|
|
|
;;; works, we need to fix all previous passes to eliminate this
|
|
|
|
;;; whole primcall business.
|
|
|
|
|
|
|
|
(define (remove-primcalls x)
|
|
|
|
;;;
|
|
|
|
(define who 'remove-primcalls)
|
|
|
|
;;;
|
|
|
|
(define (check-gensym x)
|
|
|
|
(unless (gensym? x)
|
|
|
|
(error who "invalid gensym ~s" x)))
|
|
|
|
;;;
|
|
|
|
(define (check-label x)
|
|
|
|
(record-case x
|
|
|
|
[(code-loc label)
|
|
|
|
(check-gensym label)]
|
|
|
|
[else (error who "invalid label ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (check-var x)
|
|
|
|
(record-case x
|
|
|
|
[(var) (void)]
|
|
|
|
[else (error who "invalid var ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (check-closure x)
|
|
|
|
(record-case x
|
|
|
|
[(closure label free*)
|
|
|
|
(check-label label)
|
|
|
|
(for-each check-var free*)]
|
|
|
|
[else (error who "invalid closure ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Expr x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) x]
|
|
|
|
[(var) x]
|
|
|
|
[(primref) x]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(make-bind lhs* (map Expr rhs*) (Expr body))]
|
|
|
|
[(fix lhs* rhs* body)
|
|
|
|
(make-fix lhs* rhs* (Expr body))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (Expr e0) (Expr e1) (Expr e2))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (Expr e0) (Expr e1))]
|
|
|
|
[(closure) x]
|
|
|
|
[(primcall op arg*)
|
2007-02-11 17:51:42 -05:00
|
|
|
(make-funcall (make-primref op) (map Expr arg*))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(forcall op arg*)
|
|
|
|
(make-forcall op (map Expr arg*))]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(make-funcall (Expr rator) (map Expr arg*))]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(make-jmpcall label (Expr rator) (map Expr arg*))]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(error 'new-cogen "appcall not supported yet")
|
|
|
|
(make-appcall (Expr rator) (map Expr arg*))]
|
|
|
|
[(mvcall rator k)
|
|
|
|
(make-mvcall (Expr rator) (Clambda k))]
|
|
|
|
[else (error who "invalid expr ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(make-clambda-case info (Expr body))]
|
|
|
|
[else (error who "invalid clambda-case ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(make-clambda label (map ClambdaCase case*) free*)]
|
|
|
|
[else (error who "invalid clambda ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(make-codes (map Clambda code*) (Expr body))]
|
|
|
|
[else (error who "invalid program ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
(define (eliminate-fix x)
|
|
|
|
;;;
|
|
|
|
(define who 'eliminate-fix)
|
|
|
|
;;;
|
|
|
|
(define (Expr cpvar free*)
|
|
|
|
;;;
|
|
|
|
(define (Var x)
|
|
|
|
(let f ([free* free*] [i 0])
|
|
|
|
(cond
|
|
|
|
[(null? free*) x]
|
|
|
|
[(eq? x (car free*))
|
|
|
|
(make-primcall 'cpref (list cpvar (make-constant i)))]
|
|
|
|
[else (f (cdr free*) (fxadd1 i))])))
|
|
|
|
;;;
|
|
|
|
(define (make-closure x)
|
|
|
|
(record-case x
|
|
|
|
[(closure code free*)
|
|
|
|
(cond
|
|
|
|
[(null? free*) x]
|
|
|
|
[else
|
|
|
|
(make-primcall 'make-closure
|
|
|
|
(list code (make-constant (length free*))))])]))
|
|
|
|
;;;
|
|
|
|
(define (closure-sets var x ac)
|
|
|
|
(record-case x
|
|
|
|
[(closure code free*)
|
|
|
|
(let f ([i 0] [free* free*])
|
|
|
|
(cond
|
|
|
|
[(null? free*) ac]
|
|
|
|
[else
|
|
|
|
(make-seq
|
|
|
|
(make-primcall 'closure-set!
|
|
|
|
(list var (make-constant i)
|
|
|
|
(Var (car free*))))
|
|
|
|
(f (fxadd1 i) (cdr free*)))]))]))
|
|
|
|
;;;
|
|
|
|
(define (do-fix lhs* rhs* body)
|
|
|
|
(make-bind
|
|
|
|
lhs* (map make-closure rhs*)
|
|
|
|
(let f ([lhs* lhs*] [rhs* rhs*])
|
|
|
|
(cond
|
|
|
|
[(null? lhs*) body]
|
|
|
|
[else
|
|
|
|
(closure-sets (car lhs*) (car rhs*)
|
|
|
|
(f (cdr lhs*) (cdr rhs*)))]))))
|
|
|
|
;;;
|
|
|
|
(define (Expr x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) x]
|
|
|
|
[(var) (Var x)]
|
|
|
|
[(primref) x]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(make-bind lhs* (map Expr rhs*) (Expr body))]
|
|
|
|
[(fix lhs* rhs* body)
|
|
|
|
(do-fix lhs* rhs* (Expr body))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (Expr e0) (Expr e1) (Expr e2))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (Expr e0) (Expr e1))]
|
|
|
|
[(closure)
|
|
|
|
(let ([t (unique-var 'tmp)])
|
|
|
|
(Expr (make-fix (list t) (list x) t)))]
|
|
|
|
[(primcall op arg*)
|
|
|
|
(make-appcall (make-primref op) (map Expr arg*))]
|
|
|
|
[(forcall op arg*)
|
|
|
|
(make-forcall op (map Expr arg*))]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(make-funcall (Expr rator) (map Expr arg*))]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(make-jmpcall label (Expr rator) (map Expr arg*))]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(error who "appcall not supported yet")
|
|
|
|
(make-appcall (Expr rator) (map Expr arg*))]
|
|
|
|
[(mvcall rator k)
|
|
|
|
(make-mvcall (Expr rator) (Clambda k))]
|
|
|
|
[else (error who "invalid expr ~s" x)]))
|
|
|
|
Expr)
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase free*)
|
|
|
|
(lambda (x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(record-case info
|
|
|
|
[(case-info label args proper)
|
|
|
|
(let ([cp (unique-var 'cp)])
|
|
|
|
(make-clambda-case
|
|
|
|
(make-case-info label (cons cp args) proper)
|
|
|
|
((Expr cp free*) body)))])]
|
|
|
|
[else (error who "invalid clambda-case ~s" x)])))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(make-clambda label (map (ClambdaCase free*) case*)
|
|
|
|
free*)]
|
|
|
|
[else (error who "invalid clambda ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(make-codes (map Clambda code*) ((Expr #f '()) body))]
|
|
|
|
[else (error who "invalid program ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
(define (specify-representation x)
|
|
|
|
(define who 'specify-representation)
|
|
|
|
;;;
|
|
|
|
(define fixnum-scale 4)
|
|
|
|
;;;
|
|
|
|
(define nop (make-primcall 'nop '()))
|
|
|
|
;;;
|
|
|
|
(define (constant-rep x)
|
|
|
|
(let ([c (constant-value x)])
|
|
|
|
(cond
|
|
|
|
[(fixnum? c) (make-constant (* c fixnum-scale))]
|
|
|
|
[(boolean? c) (make-constant (if c bool-t bool-f))]
|
|
|
|
[(eq? c (void)) (make-constant void-object)]
|
|
|
|
[(bwp-object? c) (make-constant bwp-object)]
|
|
|
|
[(char? c) (make-constant
|
|
|
|
(fxlogor char-tag
|
|
|
|
(fxsll (char->integer c) char-shift)))]
|
|
|
|
[(null? c) (make-constant nil)]
|
|
|
|
[else (make-constant (make-object c))])))
|
|
|
|
;;;
|
|
|
|
(define (Effect x)
|
|
|
|
(record-case x
|
|
|
|
[(constant c) nop]
|
|
|
|
[(var) nop]
|
|
|
|
[(primref) nop]
|
|
|
|
[(closure code free*) nop]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(make-bind lhs* (map Value rhs*) (Effect body))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (Pred e0) (Effect e1) (Effect e2))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (Effect e0) (Effect e1))]
|
|
|
|
[(primcall op arg*)
|
|
|
|
(case op
|
|
|
|
[(closure-set!)
|
|
|
|
(let ([x (Value (car arg*))]
|
|
|
|
[i (cadr arg*)]
|
|
|
|
[v (Value (caddr arg*))])
|
|
|
|
(record-case i
|
|
|
|
[(constant i)
|
|
|
|
(unless (fixnum? i) (err x))
|
|
|
|
(make-primcall 'mset!
|
|
|
|
(list x
|
|
|
|
(make-constant
|
|
|
|
(+ (* i wordsize)
|
|
|
|
(- disp-closure-data closure-tag)))
|
|
|
|
v))]
|
|
|
|
[else (err x)]))]
|
|
|
|
[else (error who "invalid effect prim ~s" op)])]
|
|
|
|
[(forcall op arg*)
|
|
|
|
(error who "effect forcall not supported" op)]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(make-funcall (Value rator) (map Value arg*))]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(make-jmpcall label (Value rator) (map Value arg*))]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(error who "appcall not supported yet")]
|
|
|
|
[(mvcall rator x)
|
|
|
|
(make-mvcall (Value rator) (Clambda x Effect))]
|
|
|
|
[else (error who "invalid pred expr ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Pred x)
|
|
|
|
(record-case x
|
|
|
|
[(constant c) (make-constant (if c #t #f))]
|
|
|
|
[(var) (make-primcall '!= (list x (make-constant bool-f)))]
|
|
|
|
[(primref) (make-constant #t)]
|
|
|
|
[(closure code free*) (make-constant #t)]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(make-bind lhs* (map Value rhs*) (Pred body))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (Pred e0) (Pred e1) (Pred e2))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (Effect e0) (Pred e1))]
|
|
|
|
[(primcall op arg*)
|
|
|
|
(error who "pred prim ~a not supported" op)]
|
|
|
|
[(forcall op arg*)
|
|
|
|
(error who "pred forcall not supported" op)]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(make-primcall '!=
|
|
|
|
(list (make-funcall (Value rator) (map Value arg*))
|
|
|
|
(make-constant bool-f)))]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(make-primcall '!=
|
|
|
|
(list (make-jmpcall label (Value rator) (map Value arg*))
|
|
|
|
(make-constant bool-f)))]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(error who "appcall not supported yet")]
|
|
|
|
[(mvcall rator x)
|
|
|
|
(make-mvcall (Value rator) (Clambda x Pred))]
|
|
|
|
[else (error who "invalid pred expr ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (err x)
|
|
|
|
(error who "invalid form ~s" (unparse x)))
|
|
|
|
;;;
|
|
|
|
(define (Value x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) (constant-rep x)]
|
|
|
|
[(var) x]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(primref name)
|
|
|
|
(make-primcall 'mref
|
|
|
|
(list
|
|
|
|
(make-constant (make-object name))
|
|
|
|
(make-constant
|
|
|
|
(- disp-symbol-system-value symbol-tag))))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(code-loc) (make-constant x)]
|
|
|
|
[(closure) (make-constant x)]
|
|
|
|
[(bind lhs* rhs* body)
|
|
|
|
(make-bind lhs* (map Value rhs*) (Value body))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (Pred e0) (Value e1) (Value e2))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (Effect e0) (Value e1))]
|
|
|
|
[(primcall op arg*)
|
|
|
|
(case op
|
|
|
|
[(make-closure)
|
|
|
|
(let ([label (car arg*)] [len (cadr arg*)])
|
|
|
|
(record-case len
|
|
|
|
[(constant i)
|
|
|
|
(unless (fixnum? i) (err x))
|
|
|
|
(let ([t (unique-var 't)])
|
|
|
|
(make-bind (list t)
|
|
|
|
(list (make-primcall 'alloc
|
|
|
|
(list (make-constant
|
|
|
|
(align
|
|
|
|
(+ disp-closure-data
|
|
|
|
(* i wordsize))))
|
|
|
|
(make-constant closure-tag))))
|
|
|
|
(make-seq
|
|
|
|
(make-primcall 'mset!
|
|
|
|
(list t
|
|
|
|
(make-constant (- disp-closure-code closure-tag))
|
|
|
|
(Value label)))
|
|
|
|
t)))]
|
|
|
|
[else (err x)]))]
|
|
|
|
[(cpref)
|
|
|
|
(let ([a0 (car arg*)] [a1 (cadr arg*)])
|
|
|
|
(record-case a1
|
|
|
|
[(constant i)
|
|
|
|
(unless (fixnum? i) (err x))
|
|
|
|
(make-primcall 'mem
|
|
|
|
(list (Value a0)
|
|
|
|
(make-constant
|
|
|
|
(+ (- disp-closure-data closure-tag)
|
|
|
|
(* i wordsize) ))))]
|
|
|
|
[else (err x)]))]
|
|
|
|
[else (error who "value prim ~a not supported" (unparse x))])]
|
|
|
|
[(forcall op arg*)
|
|
|
|
(error who "value forcall not supported" op)]
|
|
|
|
[(funcall rator arg*)
|
|
|
|
(make-funcall (Value rator) (map Value arg*))]
|
|
|
|
[(jmpcall label rator arg*)
|
|
|
|
(make-jmpcall label (Value rator) (map Value arg*))]
|
|
|
|
[(appcall rator arg*)
|
|
|
|
(error who "appcall not supported yet")]
|
|
|
|
[(mvcall rator x)
|
|
|
|
(make-mvcall (Value rator) (Clambda x Value))]
|
|
|
|
[else (error who "invalid value expr ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x k)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(make-clambda-case info (k body))]
|
|
|
|
[else (error who "invalid clambda-case ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x k)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(make-clambda label
|
|
|
|
(map (lambda (x) (ClambdaCase x k)) case*)
|
|
|
|
free*)]
|
|
|
|
[else (error who "invalid clambda ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(make-codes
|
|
|
|
(map (lambda (x) (Clambda x Value)) code*)
|
|
|
|
(Value body))]
|
|
|
|
[else (error who "invalid program ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
|
|
|
|
(define parameter-registers '(%edi))
|
|
|
|
(define return-value-register '%eax)
|
|
|
|
(define cp-register '%edi)
|
|
|
|
(define all-registers '(%eax %edi %ebx %edx))
|
2007-02-11 04:12:09 -05:00
|
|
|
(define argc-register '%eax)
|
2007-02-10 18:51:12 -05:00
|
|
|
|
|
|
|
(define (impose-calling-convention/evaluation-order x)
|
|
|
|
(define who 'impose-calling-convention/evaluation-order)
|
|
|
|
;;;
|
|
|
|
;;;
|
|
|
|
(define (S* x* k)
|
|
|
|
(cond
|
|
|
|
[(null? x*) (k '())]
|
|
|
|
[else
|
|
|
|
(S (car x*)
|
|
|
|
(lambda (a)
|
|
|
|
(S* (cdr x*)
|
|
|
|
(lambda (d)
|
|
|
|
(k (cons a d))))))]))
|
|
|
|
;;;
|
|
|
|
(define (S x k)
|
2007-02-11 17:23:13 -05:00
|
|
|
(cond
|
|
|
|
[(or (constant? x) (var? x))
|
|
|
|
(k x)]
|
|
|
|
[(funcall? x)
|
|
|
|
(let ([t (unique-var 'tmp)])
|
|
|
|
(do-bind (list t) (list x)
|
|
|
|
(k t)))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid S ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (do-bind lhs* rhs* body)
|
|
|
|
(cond
|
|
|
|
[(null? lhs*) body]
|
|
|
|
[else
|
|
|
|
(set! locals (cons (car lhs*) locals))
|
|
|
|
(make-seq
|
|
|
|
(V (car lhs*) (car rhs*))
|
|
|
|
(do-bind (cdr lhs*) (cdr rhs*) body))]))
|
|
|
|
;;;
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (nontail-locations args)
|
|
|
|
(let f ([regs parameter-registers] [args args])
|
|
|
|
(cond
|
|
|
|
[(null? args) (values '() '() '())]
|
|
|
|
[(null? regs) (values '() '() args)]
|
|
|
|
[else
|
|
|
|
(let-values ([(r* rl* f*) (f (cdr regs) (cdr args))])
|
|
|
|
(values (cons (car regs) r*)
|
|
|
|
(cons (car args) rl*)
|
|
|
|
f*))])))
|
2007-02-11 17:51:42 -05:00
|
|
|
(define (handle-nontail-call rator rands dest)
|
|
|
|
(let-values ([(reg-locs reg-args frm-args)
|
|
|
|
(nontail-locations (cons rator rands))])
|
|
|
|
(let ([regt* (map (lambda (x) (unique-var 'rt)) reg-args)]
|
|
|
|
[frmt* (map (lambda (x) (make-nfvar #f #f)) frm-args)])
|
|
|
|
(let ([body
|
|
|
|
(make-nframe frmt* #f
|
|
|
|
(do-bind frmt* frm-args
|
|
|
|
(do-bind regt* reg-args
|
|
|
|
(assign* reg-locs regt*
|
|
|
|
(make-seq
|
|
|
|
(make-set argc-register
|
|
|
|
(make-constant
|
|
|
|
(argc-convention (length rands))))
|
|
|
|
(make-primcall
|
|
|
|
'indirect-call
|
|
|
|
(cons argc-register
|
|
|
|
(append reg-locs frmt*))))))))])
|
|
|
|
(if dest
|
|
|
|
(make-seq body (make-set dest return-value-register))
|
|
|
|
body)))))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (V d x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) (make-set d x)]
|
|
|
|
[(var) (make-set d x)]
|
|
|
|
[(bind lhs* rhs* e)
|
|
|
|
(do-bind lhs* rhs* (V d e))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (E e0) (V d e1))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (V d e1) (V d e2))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(S* rands
|
|
|
|
(lambda (rands)
|
|
|
|
(make-set d (make-primcall op rands))))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(funcall rator rands)
|
2007-02-11 17:51:42 -05:00
|
|
|
(handle-nontail-call rator rands d)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid value ~s" x)]))
|
|
|
|
;;;
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (assign* lhs* rhs* ac)
|
|
|
|
(cond
|
|
|
|
[(null? lhs*) ac]
|
|
|
|
[else
|
|
|
|
(make-seq
|
|
|
|
(make-set (car lhs*) (car rhs*))
|
|
|
|
(assign* (cdr lhs*) (cdr rhs*) ac))]))
|
|
|
|
;;;
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (VT x)
|
|
|
|
(make-seq
|
|
|
|
(V return-value-register x)
|
2007-02-11 04:12:09 -05:00
|
|
|
(make-primcall 'return (list return-value-register))))
|
2007-02-10 18:51:12 -05:00
|
|
|
;;;
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (E x)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (E e1))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (E e1) (E e2))]
|
|
|
|
[(primcall op rands)
|
|
|
|
(S* rands
|
|
|
|
(lambda (rands)
|
|
|
|
(make-primcall op rands)))]
|
2007-02-11 17:51:42 -05:00
|
|
|
[(funcall rator rands)
|
|
|
|
(handle-nontail-call rator rands #f)]
|
2007-02-11 17:23:13 -05:00
|
|
|
[else (error who "invalid effect ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (P x)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (P e1))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (P e1) (P e2))]
|
|
|
|
[(primcall op rands)
|
|
|
|
(S* rands
|
|
|
|
(lambda (rands)
|
|
|
|
(make-primcall op rands)))]
|
|
|
|
[else (error who "invalid pred ~s" x)]))
|
|
|
|
;;;
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (Tail x)
|
|
|
|
(record-case x
|
|
|
|
[(constant) (VT x)]
|
|
|
|
[(var) (VT x)]
|
|
|
|
[(primcall) (VT x)]
|
|
|
|
[(bind lhs* rhs* e)
|
|
|
|
(do-bind lhs* rhs* (Tail e))]
|
|
|
|
[(seq e0 e1)
|
|
|
|
(make-seq (E e0) (Tail e1))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (Tail e1) (Tail e2))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(funcall rator rands)
|
|
|
|
(let ([cpt (unique-var 'rator)]
|
|
|
|
[rt* (map (lambda (x) (unique-var 't)) rands)])
|
|
|
|
(do-bind rt* rands
|
|
|
|
(do-bind (list cpt) (list rator)
|
|
|
|
(let ([args (cons cpt rt*)]
|
|
|
|
[locs (formals-locations (cons cpt rt*))])
|
|
|
|
(assign* (reverse locs)
|
|
|
|
(reverse args)
|
|
|
|
(make-seq
|
|
|
|
(make-set argc-register
|
|
|
|
(make-constant
|
|
|
|
(argc-convention (length rands))))
|
|
|
|
(make-primcall 'indirect-jump locs)))))))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid tail ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (formals-locations args)
|
|
|
|
(let f ([regs parameter-registers] [args args])
|
|
|
|
(cond
|
|
|
|
[(null? args) '()]
|
|
|
|
[(null? regs)
|
2007-02-11 04:12:09 -05:00
|
|
|
(let f ([i 1] [args args])
|
2007-02-10 18:51:12 -05:00
|
|
|
(cond
|
|
|
|
[(null? args) '()]
|
|
|
|
[else
|
2007-02-11 04:12:09 -05:00
|
|
|
(cons (mkfvar i)
|
2007-02-10 18:51:12 -05:00
|
|
|
(f (fxadd1 i) (cdr args)))]))]
|
|
|
|
[else
|
|
|
|
(cons (car regs) (f (cdr regs) (cdr args)))])))
|
|
|
|
;;;
|
|
|
|
(define locals '())
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(record-case info
|
|
|
|
[(case-info label args proper)
|
|
|
|
(set! locals args)
|
|
|
|
(let* ([locs (formals-locations args)]
|
|
|
|
[body (let f ([args args] [locs locs])
|
|
|
|
(cond
|
|
|
|
[(null? args) (Tail body)]
|
|
|
|
[else
|
|
|
|
(make-seq
|
|
|
|
(make-set (car args) (car locs))
|
|
|
|
(f (cdr args) (cdr locs)))]))])
|
|
|
|
(make-clambda-case
|
|
|
|
(make-case-info label locs proper)
|
|
|
|
(make-locals locals body)))])]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(make-clambda label (map ClambdaCase case*) free*)]))
|
|
|
|
;;;
|
|
|
|
(define (Main x)
|
|
|
|
(set! locals '())
|
|
|
|
(let ([x (Tail x)])
|
|
|
|
(make-locals locals x)))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(make-codes (map Clambda code*) (Main body))]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
|
|
|
|
(module ListyGraphs
|
|
|
|
(empty-graph add-edge! empty-graph? print-graph node-neighbors
|
|
|
|
delete-node!)
|
|
|
|
;;;
|
|
|
|
(define-record graph (ls))
|
|
|
|
;;;
|
|
|
|
(define (empty-graph) (make-graph '()))
|
|
|
|
;;;
|
|
|
|
(define (empty-graph? g)
|
|
|
|
(andmap (lambda (x) (null? (cdr x))) (graph-ls g)))
|
|
|
|
;;;
|
|
|
|
(define (add-edge! g x y)
|
|
|
|
(let ([ls (graph-ls g)])
|
|
|
|
(cond
|
|
|
|
[(assq x ls) =>
|
|
|
|
(lambda (p0)
|
|
|
|
(unless (memq y (cdr p0))
|
2007-02-11 04:12:09 -05:00
|
|
|
(set-cdr! p0 (cons y (cdr p0)))
|
|
|
|
(cond
|
|
|
|
[(assq y ls) =>
|
|
|
|
(lambda (p1)
|
|
|
|
(set-cdr! p1 (cons x (cdr p1))))]
|
|
|
|
[else
|
|
|
|
(set-graph-ls! g
|
|
|
|
(cons (list y x) ls))])))]
|
|
|
|
[(assq y ls) =>
|
|
|
|
(lambda (p1)
|
|
|
|
(set-cdr! p1 (cons x (cdr p1)))
|
|
|
|
(set-graph-ls! g (cons (list x y) ls)))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else
|
|
|
|
(set-graph-ls! g
|
|
|
|
(list* (list x y)
|
|
|
|
(list y x)
|
|
|
|
ls))])))
|
|
|
|
(define (print-graph g)
|
|
|
|
(printf "G={\n")
|
|
|
|
(parameterize ([print-gensym 'pretty])
|
|
|
|
(for-each (lambda (x)
|
|
|
|
(let ([lhs (car x)] [rhs* (cdr x)])
|
|
|
|
(printf " ~s => ~s\n"
|
|
|
|
(unparse lhs)
|
|
|
|
(map unparse rhs*))))
|
|
|
|
(graph-ls g)))
|
|
|
|
(printf "}\n"))
|
|
|
|
(define (node-neighbors x g)
|
|
|
|
(cond
|
|
|
|
[(assq x (graph-ls g)) => cdr]
|
|
|
|
[else '()]))
|
|
|
|
(define (delete-node! x g)
|
|
|
|
(let ([ls (graph-ls g)])
|
|
|
|
(cond
|
|
|
|
[(assq x ls) =>
|
|
|
|
(lambda (p)
|
|
|
|
(for-each (lambda (y)
|
|
|
|
(let ([p (assq y ls)])
|
|
|
|
(set-cdr! p (set-rem x (cdr p)))))
|
|
|
|
(cdr p))
|
|
|
|
(set-cdr! p '()))]
|
|
|
|
[else (void)])))
|
|
|
|
;;;
|
|
|
|
#|ListyGraphs|#)
|
|
|
|
|
|
|
|
(define (set-add x s)
|
|
|
|
(cond
|
|
|
|
[(memq x s) s]
|
|
|
|
[else (cons x s)]))
|
|
|
|
|
|
|
|
(define (set-rem x s)
|
|
|
|
(cond
|
|
|
|
[(null? s) '()]
|
|
|
|
[(eq? x (car s)) (cdr s)]
|
|
|
|
[else (cons (car s) (set-rem x (cdr s)))]))
|
|
|
|
|
|
|
|
(define (set-difference s1 s2)
|
|
|
|
(cond
|
|
|
|
[(null? s2) s1]
|
|
|
|
[else (set-difference (set-rem (car s2) s1) (cdr s2))]))
|
|
|
|
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (set-union s1 s2)
|
|
|
|
(cond
|
|
|
|
[(null? s1) s2]
|
|
|
|
[(memq (car s1) s2) (set-union (cdr s1) s2)]
|
|
|
|
[else (cons (car s1) (set-union (cdr s1) s2))]))
|
|
|
|
|
2007-02-11 04:12:09 -05:00
|
|
|
|
2007-02-10 18:51:12 -05:00
|
|
|
(module (color-by-chaitin)
|
|
|
|
(import ListyGraphs)
|
|
|
|
;;;
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (build-graph x reg?)
|
2007-02-10 18:51:12 -05:00
|
|
|
(define who 'build-graph)
|
|
|
|
(define g (empty-graph))
|
|
|
|
(define (add-rands ls s)
|
|
|
|
(cond
|
|
|
|
[(null? ls) s]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(or (reg? (car ls)) (var? (car ls)) (nfvar? (car ls)))
|
2007-02-10 18:51:12 -05:00
|
|
|
(add-rands (cdr ls) (set-add (car ls) s))]
|
|
|
|
[else (add-rands (cdr ls) s)]))
|
|
|
|
(define (Rhs x s)
|
|
|
|
(record-case x
|
|
|
|
[(primcall op rand*) (add-rands rand* s)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[else
|
|
|
|
(if (or (var? x) (reg? x) (nfvar? x))
|
|
|
|
(set-add x s)
|
|
|
|
s)]))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (E x s)
|
|
|
|
(record-case x
|
2007-02-11 04:12:09 -05:00
|
|
|
[(set lhs rhs)
|
|
|
|
(cond
|
|
|
|
[(or (var? lhs) (reg? lhs))
|
|
|
|
(cond
|
|
|
|
[(or (var? rhs) (reg? rhs))
|
|
|
|
(let ([s (set-rem rhs (set-rem lhs s))])
|
|
|
|
(for-each (lambda (x) (add-edge! g lhs x)) s)
|
|
|
|
(cons rhs s))]
|
|
|
|
[else
|
|
|
|
(let ([s (set-rem lhs s)])
|
|
|
|
(for-each (lambda (x) (add-edge! g lhs x)) s)
|
|
|
|
(Rhs rhs s))])]
|
|
|
|
[(nfvar? lhs)
|
|
|
|
(let ([s (set-rem lhs s)])
|
|
|
|
(set-nfvar-conf! lhs s)
|
|
|
|
(Rhs rhs s))]
|
|
|
|
[else (Rhs rhs s)])]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(seq e0 e1) (E e0 (E e1 s))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(let ([s1 (E e1 s)] [s2 (E e2 s)])
|
|
|
|
(P e0 s1 s2 (set-union s1 s2)))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands) (add-rands rands s)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(nframe vars live body)
|
|
|
|
(when (reg? return-value-register)
|
|
|
|
(for-each
|
|
|
|
(lambda (x)
|
|
|
|
(for-each (lambda (r)
|
|
|
|
(add-edge! g x r))
|
|
|
|
all-registers))
|
|
|
|
s))
|
|
|
|
(set-nframe-live! x s)
|
|
|
|
(E body s)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid effect ~s" x)]))
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (P x st sf su)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1)
|
|
|
|
(E e0 (P e1 st sf su))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(let ([s1 (P e1 st sf su)] [s2 (P e2 st sf su)])
|
|
|
|
(P e0 s1 s2 (set-union s1 s2)))]
|
|
|
|
[(primcall op rands)
|
|
|
|
(add-rands rands su)]
|
|
|
|
[else (error who "invalid pred ~s" x)]))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (T x)
|
|
|
|
(record-case x
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(let ([s1 (T e1)] [s2 (T e2)])
|
|
|
|
(P e0 s1 s2 (set-union s1 s2)))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(add-rands rands '())]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(seq e0 e1) (E e0 (T e1))]
|
|
|
|
[else (error who "invalid tail ~s" x)]))
|
|
|
|
(let ([s (T x)])
|
|
|
|
;(print-graph g)
|
|
|
|
g))
|
|
|
|
;;;
|
|
|
|
(define (color-graph sp* un* g)
|
|
|
|
(define (find-low-degree ls g)
|
|
|
|
(cond
|
|
|
|
[(null? ls) #f]
|
|
|
|
[(fx< (length (node-neighbors (car ls) g))
|
|
|
|
(length all-registers))
|
|
|
|
(car ls)]
|
|
|
|
[else (find-low-degree (cdr ls) g)]))
|
|
|
|
(define (find-color/maybe x confs env)
|
|
|
|
(let ([cr (map (lambda (x)
|
|
|
|
(cond
|
2007-02-11 04:12:09 -05:00
|
|
|
[(symbol? x) x]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(assq x env) => cdr]
|
|
|
|
[else #f]))
|
|
|
|
confs)])
|
|
|
|
(let ([r* (set-difference all-registers cr)])
|
|
|
|
(if (null? r*)
|
|
|
|
#f
|
|
|
|
(car r*)))))
|
|
|
|
(define (find-color x confs env)
|
|
|
|
(or (find-color/maybe x confs env)
|
|
|
|
(error 'find-color "cannot find color for ~s" x)))
|
|
|
|
(cond
|
|
|
|
[(and (null? sp*) (null? un*)) (values '() '() '())]
|
|
|
|
[(find-low-degree un* g) =>
|
|
|
|
(lambda (un)
|
|
|
|
(let ([n* (node-neighbors un g)])
|
|
|
|
(delete-node! un g)
|
|
|
|
(let-values ([(spills sp* env)
|
|
|
|
(color-graph sp* (set-rem un un*) g)])
|
|
|
|
(let ([r (find-color un n* env)])
|
|
|
|
(values spills sp*
|
|
|
|
(cons (cons un r) env))))))]
|
|
|
|
[(find-low-degree sp* g) =>
|
|
|
|
(lambda (sp)
|
|
|
|
(let ([n* (node-neighbors sp g)])
|
|
|
|
(delete-node! sp g)
|
|
|
|
(let-values ([(spills sp* env)
|
|
|
|
(color-graph (set-rem sp sp*) un* g)])
|
|
|
|
(let ([r (find-color sp n* env)])
|
|
|
|
(values spills (cons sp sp*)
|
|
|
|
(cons (cons sp r) env))))))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(pair? sp*)
|
|
|
|
(let ([sp (car sp*)])
|
|
|
|
(let ([n* (node-neighbors sp g)])
|
|
|
|
(delete-node! sp g)
|
|
|
|
(let-values ([(spills sp* env)
|
|
|
|
(color-graph (set-rem sp sp*) un* g)])
|
|
|
|
(let ([r (find-color/maybe sp n* env)])
|
|
|
|
(if r
|
|
|
|
(values spills (cons sp sp*)
|
|
|
|
(cons (cons sp r) env))
|
|
|
|
(values (cons sp spills) sp* env))))))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error color-graph "whoaaa")]))
|
|
|
|
;;;
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (substitute env x)
|
2007-02-10 18:51:12 -05:00
|
|
|
(define who 'substitute)
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (max-live vars i)
|
|
|
|
(cond
|
|
|
|
[(null? vars) i]
|
|
|
|
[else (max-live (cdr vars)
|
|
|
|
(record-case (car vars)
|
|
|
|
[(fvar j) (max i j)]
|
|
|
|
[else i]))]))
|
|
|
|
(define (actual-frame-size vars i)
|
|
|
|
(define (conflicts? i ls)
|
|
|
|
(and (not (null? ls))
|
2007-02-11 17:34:13 -05:00
|
|
|
(or (let f ([x (car ls)])
|
|
|
|
(record-case x
|
|
|
|
[(fvar j) (eq? i j)]
|
2007-02-11 17:51:42 -05:00
|
|
|
[(var)
|
|
|
|
(cond
|
|
|
|
[(assq x env) => (lambda (x) (f (cdr x)))]
|
|
|
|
[else #f])]
|
2007-02-11 17:34:13 -05:00
|
|
|
[(nfvar conf loc) (f loc)]
|
|
|
|
[else #f]))
|
2007-02-11 04:12:09 -05:00
|
|
|
(conflicts? i (cdr ls)))))
|
|
|
|
(define (frame-size-ok? i vars)
|
|
|
|
(or (null? vars)
|
2007-02-11 17:34:13 -05:00
|
|
|
(and (not (conflicts? i (nfvar-conf (car vars))))
|
2007-02-11 04:12:09 -05:00
|
|
|
(frame-size-ok? (fxadd1 i) (cdr vars)))))
|
|
|
|
(cond
|
|
|
|
[(frame-size-ok? i vars) i]
|
|
|
|
[else (actual-frame-size vars (fxadd1 i))]))
|
|
|
|
(define (assign-frame-vars! vars i)
|
|
|
|
(unless (null? vars)
|
|
|
|
(set-nfvar-loc! (car vars) (mkfvar i))
|
|
|
|
(assign-frame-vars! (cdr vars) (fxadd1 i))))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (Var x)
|
|
|
|
(cond
|
|
|
|
[(assq x env) => cdr]
|
2007-02-11 04:12:09 -05:00
|
|
|
[else x]))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (Rhs x)
|
|
|
|
(record-case x
|
|
|
|
[(var) (Var x)]
|
|
|
|
[(primcall op rand*)
|
|
|
|
(make-primcall op (map Rand rand*))]
|
|
|
|
[else x]))
|
|
|
|
(define (Rand x)
|
|
|
|
(record-case x
|
|
|
|
[(var) (Var x)]
|
|
|
|
[else x]))
|
|
|
|
(define (Lhs x)
|
|
|
|
(record-case x
|
|
|
|
[(var) (Var x)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(nfvar confs loc)
|
|
|
|
(or loc (error who "LHS not set ~s" x))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else x]))
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (NFE idx x)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (NFE idx e1))]
|
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
|
|
|
[(indirect-call)
|
|
|
|
(make-primcall op
|
|
|
|
(cons (make-constant idx) (map Rand rands)))]
|
|
|
|
[else (error who "invalid NFE ~s" x)])]
|
|
|
|
[else (error who "invalid NF effect ~s" x)]))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (E x)
|
|
|
|
(record-case x
|
|
|
|
[(set lhs rhs)
|
|
|
|
(let ([lhs (Lhs lhs)] [rhs (Rhs rhs)])
|
|
|
|
(cond
|
2007-02-11 04:12:09 -05:00
|
|
|
[(or (eq? lhs rhs)
|
|
|
|
(and (fvar? lhs) (fvar? rhs)
|
|
|
|
(fixnum? (fvar-idx lhs))
|
|
|
|
(fixnum? (fvar-idx rhs))
|
|
|
|
(fx= (fvar-idx lhs) (fvar-idx rhs))))
|
|
|
|
(make-primcall 'nop '())]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (make-set lhs rhs)]))]
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (E e1))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (E e1) (E e2))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands)
|
2007-02-11 04:12:09 -05:00
|
|
|
(make-primcall op (map Rand rands))]
|
|
|
|
[(nframe vars live body)
|
|
|
|
;;; 1 is for the rp address
|
|
|
|
;(printf "live=~s\n" live)
|
|
|
|
(let ([i (actual-frame-size vars
|
|
|
|
(fx+ 2 (max-live (map Lhs live) 0)))])
|
|
|
|
(assign-frame-vars! vars i)
|
|
|
|
(NFE (fxsub1 i) body))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid effect ~s" x)]))
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (P x)
|
|
|
|
(record-case x
|
|
|
|
[(primcall op rands)
|
|
|
|
(make-primcall op (map Rand rands))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (P e1) (P e2))]
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (P e1))]
|
|
|
|
[else (error who "invalid pred ~s" x)]))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (T x)
|
|
|
|
(record-case x
|
2007-02-11 04:12:09 -05:00
|
|
|
[(primcall op rands) x]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (T e1) (T e2))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(seq e0 e1) (make-seq (E e0) (T e1))]
|
|
|
|
[else (error who "invalid tail ~s" x)]))
|
2007-02-11 17:34:13 -05:00
|
|
|
;(print-code x)
|
2007-02-10 18:51:12 -05:00
|
|
|
(T x))
|
|
|
|
;;;
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (do-spill sp* g)
|
|
|
|
(define (find/set-loc x)
|
|
|
|
(let ([ls (node-neighbors x g)])
|
|
|
|
(define (conflicts? i ls)
|
|
|
|
(and (pair? ls)
|
|
|
|
(or (record-case (car ls)
|
|
|
|
[(fvar j)
|
|
|
|
(and (fixnum? j) (fx= i j))]
|
|
|
|
[else #f])
|
|
|
|
(conflicts? i (cdr ls)))))
|
|
|
|
(let f ([i 1])
|
|
|
|
(cond
|
|
|
|
[(conflicts? i ls) (f (fxadd1 i))]
|
|
|
|
[else
|
|
|
|
(let ([fv (mkfvar i)])
|
|
|
|
(for-each (lambda (y) (add-edge! g y fv)) ls)
|
|
|
|
(delete-node! x g)
|
|
|
|
(cons x fv))]))))
|
|
|
|
(map find/set-loc sp*))
|
|
|
|
;;;
|
|
|
|
(define (add-unspillables un* x)
|
|
|
|
(define who 'add-unspillables)
|
|
|
|
(define (mku)
|
|
|
|
(let ([u (unique-var 'u)])
|
|
|
|
(set! un* (cons u un*))
|
|
|
|
u))
|
|
|
|
(define (S* ls k)
|
|
|
|
(cond
|
|
|
|
[(null? ls) (k '())]
|
|
|
|
[else
|
|
|
|
(let ([a (car ls)])
|
|
|
|
(S* (cdr ls)
|
|
|
|
(lambda (d)
|
|
|
|
(cond
|
|
|
|
[(fvar? a)
|
|
|
|
(let ([u (mku)])
|
|
|
|
(make-seq
|
|
|
|
(make-set u a)
|
|
|
|
(k (cons u d))))]
|
|
|
|
[else (k (cons a d))]))))]))
|
|
|
|
(define (E x)
|
|
|
|
(record-case x
|
|
|
|
[(set lhs rhs)
|
|
|
|
(cond
|
|
|
|
[(or (constant? rhs) (var? rhs) (symbol? rhs)) x]
|
|
|
|
[(fvar? lhs)
|
|
|
|
(cond
|
|
|
|
[else
|
|
|
|
(let ([u (mku)])
|
|
|
|
(make-seq
|
|
|
|
(E (make-set u rhs))
|
|
|
|
(make-set lhs u)))])]
|
|
|
|
[(fvar? rhs) x]
|
|
|
|
[(primcall? rhs)
|
|
|
|
(S* (primcall-arg* rhs)
|
|
|
|
(lambda (s*)
|
|
|
|
(make-set lhs
|
|
|
|
(make-primcall (primcall-op rhs) s*))))]
|
|
|
|
[else (error who "invalid set in ~s" x)])]
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (E e1))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (E e1) (E e2))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
|
|
|
[(nop) x]
|
|
|
|
[(indirect-call) x]
|
|
|
|
[else (error who "invalid op in ~s" x)])]
|
|
|
|
[else (error who "invalid effect ~s" x)]))
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (P x)
|
|
|
|
(record-case x
|
|
|
|
[(primcall op rands)
|
|
|
|
(let ([a0 (car rands)] [a1 (cadr rands)])
|
|
|
|
(cond
|
|
|
|
[(and (fvar? a0) (fvar? a1))
|
|
|
|
(let ([u (mku)])
|
|
|
|
(make-seq
|
|
|
|
(make-set u a0)
|
|
|
|
(make-primcall op u a1)))]
|
|
|
|
[else x]))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (P e1) (P e2))]
|
|
|
|
[(seq e0 e1) (make-seq (E e0) (P e1))]
|
|
|
|
[else (error who "invalid pred ~s" x)]))
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (T x)
|
|
|
|
(record-case x
|
|
|
|
[(primcall op rands) x]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(make-conditional (P e0) (T e1) (T e2))]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(seq e0 e1) (make-seq (E e0) (T e1))]
|
|
|
|
[else (error who "invalid tail ~s" x)]))
|
|
|
|
(let ([x (T x)])
|
|
|
|
(values un* x)))
|
2007-02-10 18:51:12 -05:00
|
|
|
;;;
|
|
|
|
(define (color-program x)
|
|
|
|
(define who 'color-program)
|
|
|
|
(record-case x
|
|
|
|
[(locals sp* body)
|
2007-02-11 04:12:09 -05:00
|
|
|
(let ([frame-g (build-graph body fvar?)])
|
|
|
|
(let loop ([sp* sp*] [un* '()] [body body])
|
|
|
|
(let ([g (build-graph body symbol?)])
|
|
|
|
; (printf "loop:\n")
|
|
|
|
; (print-code body)
|
|
|
|
(let-values ([(spills sp* env) (color-graph sp* un* g)])
|
|
|
|
(cond
|
2007-02-11 17:23:13 -05:00
|
|
|
[(null? spills) (substitute env body)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[else
|
|
|
|
(let* ([env (do-spill spills frame-g)]
|
2007-02-11 17:23:13 -05:00
|
|
|
[body (substitute env body)])
|
2007-02-11 04:12:09 -05:00
|
|
|
(let-values ([(un* body)
|
|
|
|
(add-unspillables un* body)])
|
|
|
|
(loop sp* un* body)))])))))]))
|
2007-02-10 18:51:12 -05:00
|
|
|
;;;
|
|
|
|
(define (color-by-chaitin x)
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(make-clambda-case info (color-program body))]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda label case* free*)
|
|
|
|
(make-clambda label (map ClambdaCase case*) free*)]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(make-codes (map Clambda code*) (color-program body))]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
#|chaitin module|#)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
(define (flatten-codes x)
|
|
|
|
(define who 'flatten-codes)
|
|
|
|
;;;
|
2007-02-11 04:12:09 -05:00
|
|
|
(define (FVar i)
|
|
|
|
`(disp ,(* i (- wordsize)) ,fpr))
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (Rand x)
|
|
|
|
(record-case x
|
|
|
|
[(constant c)
|
|
|
|
(record-case c
|
|
|
|
[(code-loc label) (label-address label)]
|
|
|
|
[(closure label free*)
|
|
|
|
(unless (null? free*)
|
|
|
|
(error who "nonempty closure"))
|
|
|
|
`(obj ,c)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(object o)
|
|
|
|
`(obj ,o)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else
|
|
|
|
(if (integer? c)
|
|
|
|
c
|
|
|
|
(error who "invalid constant rand ~s" c))])]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(fvar i) (FVar i)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
|
|
|
[(mem) `(disp . ,(map Rand rands))]
|
|
|
|
[else (error who "invalid rand ~s" x)])]
|
|
|
|
[else
|
|
|
|
(if (symbol? x)
|
|
|
|
x
|
|
|
|
(error who "invalid rand ~s" x))]))
|
|
|
|
;;;
|
|
|
|
(define (Rhs x d ac)
|
|
|
|
(record-case x
|
|
|
|
[(constant c)
|
|
|
|
(cons `(movl ,(Rand x) ,d) ac)]
|
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
2007-02-11 04:12:09 -05:00
|
|
|
[(mref)
|
2007-02-10 18:51:12 -05:00
|
|
|
(cons `(movl (disp ,(Rand (car rands))
|
|
|
|
,(Rand (cadr rands)))
|
|
|
|
,d)
|
|
|
|
ac)]
|
|
|
|
[(alloc)
|
|
|
|
(let ([sz (Rand (car rands))]
|
|
|
|
[tag (Rand (cadr rands))])
|
|
|
|
(list* `(movl ,apr ,d)
|
|
|
|
`(addl ,tag ,d)
|
|
|
|
`(addl ,sz ,apr)
|
|
|
|
ac))]
|
|
|
|
[else (error who "invalid rhs ~s" x)])]
|
|
|
|
[else
|
|
|
|
(if (symbol? x)
|
|
|
|
(cons `(movl ,x ,d) ac)
|
|
|
|
(error who "invalid rhs ~s" x))]))
|
|
|
|
;;;
|
|
|
|
(define (E x ac)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1) (E e0 (E e1 ac))]
|
|
|
|
[(set lhs rhs)
|
|
|
|
(Rhs rhs (Rand lhs) ac)]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(let ([lf (unique-label)])
|
|
|
|
(P e0 #f lf
|
|
|
|
(E e1
|
|
|
|
(cons `(jmp ,lf)
|
|
|
|
(E e2 (cons lf ac))))))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
|
|
|
[(nop) ac]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(indirect-call)
|
|
|
|
(record-case (car rands)
|
|
|
|
[(constant i)
|
|
|
|
(list* `(subl ,(* (fxsub1 i) wordsize) ,fpr)
|
|
|
|
`(call (disp ,(fx- disp-closure-code closure-tag) ,cp-register))
|
|
|
|
`(addl ,(* (fxsub1 i) wordsize) ,fpr)
|
|
|
|
ac)]
|
|
|
|
[else (error who "invalid ~s" x)])]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid effect ~s" x)])]
|
|
|
|
[else (error who "invalid effect ~s" x)]))
|
|
|
|
;;;
|
2007-02-11 17:23:13 -05:00
|
|
|
(define (unique-label)
|
|
|
|
(label (gensym)))
|
|
|
|
;;;
|
|
|
|
(define (P x lt lf ac)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1)
|
|
|
|
(E e0 (P e1 lt lf ac))]
|
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(cond
|
|
|
|
[(and lt lf)
|
|
|
|
(let ([l (unique-label)])
|
|
|
|
(P e0 #f l
|
|
|
|
(P e1 lt lf
|
|
|
|
(cons l (P e2 lt lf ac)))))]
|
|
|
|
[lt
|
|
|
|
(let ([lf (unique-label)] [l (unique-label)])
|
|
|
|
(P e0 #f l
|
|
|
|
(P e1 lt lf
|
|
|
|
(cons l (P e2 lt #f (cons lf ac))))))]
|
|
|
|
[lf
|
|
|
|
(let ([lt (unique-label)] [l (unique-label)])
|
|
|
|
(P e0 #f l
|
|
|
|
(P e1 lt lf
|
|
|
|
(cons l (P e2 #f lf (cons lt ac))))))]
|
|
|
|
[else
|
|
|
|
(let ([lf (unique-label)] [l (unique-label)])
|
|
|
|
(P e0 #f l
|
|
|
|
(P e1 #f #f
|
|
|
|
(cons `(jmp ,lf)
|
|
|
|
(cons l (P e2 #f #f (cons lf ac)))))))])]
|
|
|
|
[(primcall op rands)
|
|
|
|
(let ([a0 (car rands)] [a1 (cadr rands)])
|
|
|
|
(define (notop x)
|
|
|
|
(cond
|
|
|
|
[(assq x '([= !=] [!= =] [< >=] [<= >] [> <=] [>= <]))
|
|
|
|
=> cadr]
|
|
|
|
[else (error who "invalid op ~s" x)]))
|
|
|
|
(define (jmpname x)
|
|
|
|
(cond
|
|
|
|
[(assq x '([= je] [!= jne] [< jl] [<= jle] [> jg] [>= jge]))
|
|
|
|
=> cadr]
|
|
|
|
[else (error who "invalid jmpname ~s" x)]))
|
|
|
|
(define (revjmpname x)
|
|
|
|
(cond
|
|
|
|
[(assq x '([= je] [!= jne] [< jg] [<= jge] [> jl] [>= jle]))
|
|
|
|
=> cadr]
|
|
|
|
[else (error who "invalid jmpname ~s" x)]))
|
|
|
|
(define (cmp op a0 a1 lab ac)
|
|
|
|
(cond
|
|
|
|
[(or (symbol? a0) (constant? a1))
|
|
|
|
(list* `(cmpl ,(Rand a1) ,(Rand a0))
|
|
|
|
`(,(jmpname op) ,lab)
|
|
|
|
ac)]
|
|
|
|
[(or (symbol? a1) (constant? a0))
|
|
|
|
(list* `(cmpl ,(Rand a0) ,(Rand a1))
|
|
|
|
`(,(revjmpname op) ,lab)
|
|
|
|
ac)]
|
|
|
|
[else (error who "invalid ops ~s ~s" a0 a1)]))
|
|
|
|
(cond
|
|
|
|
[(and lt lf)
|
|
|
|
(cmp op a0 a1 lt
|
|
|
|
(cons `(jmp ,lf) ac))]
|
|
|
|
[lt
|
|
|
|
(cmp op a0 a1 lt ac)]
|
|
|
|
[lf
|
|
|
|
(cmp (notop op) a0 a1 lf ac)]
|
|
|
|
[else ac]))]
|
|
|
|
[else (error who "invalid pred ~s" x)]))
|
|
|
|
;;;
|
2007-02-10 18:51:12 -05:00
|
|
|
(define (T x ac)
|
|
|
|
(record-case x
|
|
|
|
[(seq e0 e1) (E e0 (T e1 ac))]
|
2007-02-11 17:23:13 -05:00
|
|
|
[(conditional e0 e1 e2)
|
|
|
|
(let ([L (unique-label)])
|
|
|
|
(P e0 #f L (T e1 (cons L (T e2 ac)))))]
|
2007-02-10 18:51:12 -05:00
|
|
|
[(primcall op rands)
|
|
|
|
(case op
|
|
|
|
[(return) (cons '(ret) ac)]
|
2007-02-11 04:12:09 -05:00
|
|
|
[(indirect-jump)
|
|
|
|
(cons `(jmp (disp ,(fx- disp-closure-code closure-tag) ,cp-register))
|
|
|
|
ac)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[else (error who "invalid tail ~s" x)])]
|
|
|
|
[else (error who "invalid tail ~s" x)]))
|
|
|
|
;;;
|
|
|
|
(define (ClambdaCase x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda-case info body)
|
|
|
|
(record-case info
|
|
|
|
[(case-info L args proper)
|
|
|
|
(unless proper (error who "improper lambda"))
|
|
|
|
(cons (label L)
|
|
|
|
(T body '()))])]))
|
|
|
|
;;;
|
|
|
|
(define (Clambda x)
|
|
|
|
(record-case x
|
|
|
|
[(clambda L case* free*)
|
|
|
|
(unless (fx= (length case*) 1)
|
|
|
|
(error who "not a lambda"))
|
|
|
|
(list* (length free*)
|
|
|
|
(label L)
|
|
|
|
(ClambdaCase (car case*)))]))
|
|
|
|
;;;
|
|
|
|
(define (Program x)
|
|
|
|
(record-case x
|
|
|
|
[(codes code* body)
|
|
|
|
(cons (list* 0
|
|
|
|
(label (gensym))
|
|
|
|
(T body '()))
|
|
|
|
(map Clambda code*))]))
|
|
|
|
;;;
|
|
|
|
(Program x))
|
|
|
|
|
|
|
|
(define (print-code x)
|
2007-02-11 17:34:13 -05:00
|
|
|
(parameterize ([print-gensym '#t])
|
2007-02-10 18:51:12 -05:00
|
|
|
(pretty-print (unparse x))))
|
|
|
|
|
|
|
|
(define (alt-cogen x)
|
|
|
|
(verify-new-cogen-input x)
|
2007-02-11 17:51:42 -05:00
|
|
|
(let* (
|
|
|
|
;[foo (print-code x)]
|
|
|
|
[x (remove-primcalls x)]
|
|
|
|
;[foo (print-code x)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[x (eliminate-fix x)]
|
|
|
|
[x (specify-representation x)]
|
|
|
|
[x (impose-calling-convention/evaluation-order x)]
|
2007-02-11 04:12:09 -05:00
|
|
|
; [foo (print-code x)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[x (color-by-chaitin x)]
|
2007-02-11 04:12:09 -05:00
|
|
|
;[foo (print-code x)]
|
2007-02-10 18:51:12 -05:00
|
|
|
[ls (flatten-codes x)])
|
2007-02-11 04:12:09 -05:00
|
|
|
(parameterize ([gensym-prefix "L"]
|
|
|
|
[print-gensym #f])
|
|
|
|
(for-each
|
|
|
|
(lambda (ls)
|
|
|
|
(newline)
|
|
|
|
(for-each (lambda (x) (printf " ~s\n" x)) ls))
|
|
|
|
ls))
|
2007-02-10 18:51:12 -05:00
|
|
|
ls))
|
|
|
|
|
|
|
|
#|module alt-cogen|#)
|
|
|
|
|
|
|
|
|