Rohan Jacob-Rao
10 years ago
parent
6bde9bda88
commit
fc0ca6b6db
@ -0,0 +1,19 @@
|
|||||||
|
(*
|
||||||
|
* Copyright (c) 2015 - Facebook.
|
||||||
|
* All rights reserved.
|
||||||
|
*)
|
||||||
|
type prog = Prog of func_def list
|
||||||
|
and func_def = FuncDef of string * typ option * (typ * string) list * instr list
|
||||||
|
and typ =
|
||||||
|
| Tint of int
|
||||||
|
| Tfloat (* just one type for now *)
|
||||||
|
| Tptr of typ
|
||||||
|
| Tvector of int * typ
|
||||||
|
| Tarray of int * typ
|
||||||
|
and instr =
|
||||||
|
| Ret of (typ * value) option
|
||||||
|
and value =
|
||||||
|
| True
|
||||||
|
| False
|
||||||
|
| Intlit of int
|
||||||
|
| Null
|
||||||
@ -0,0 +1,151 @@
|
|||||||
|
(*
|
||||||
|
* Copyright (c) 2015 - Facebook.
|
||||||
|
* All rights reserved.
|
||||||
|
*)
|
||||||
|
{
|
||||||
|
open Lexing
|
||||||
|
open Parser
|
||||||
|
|
||||||
|
exception LexingError of string
|
||||||
|
}
|
||||||
|
|
||||||
|
let space = [' ' '\t']
|
||||||
|
let newline = '\n'
|
||||||
|
let comment = ';' [^ '\n']*
|
||||||
|
|
||||||
|
let nonzero_digit = ['1'-'9']
|
||||||
|
let digit = ['0'-'9']
|
||||||
|
let pos_int = nonzero_digit digit*
|
||||||
|
let intlit = '-'? digit+
|
||||||
|
|
||||||
|
let lower = ['a'-'z']
|
||||||
|
let upper = ['A'-'Z']
|
||||||
|
let id_special_char = ['-' '$' '.' '_']
|
||||||
|
let id_char = lower | upper | id_special_char
|
||||||
|
let id = ['%' '@'] id_char (id_char | digit)*
|
||||||
|
|
||||||
|
rule token = parse
|
||||||
|
| space | comment { token lexbuf }
|
||||||
|
| newline { token lexbuf }
|
||||||
|
|
||||||
|
(* keywords *)
|
||||||
|
| "define" { DEFINE }
|
||||||
|
|
||||||
|
(* delimiters *)
|
||||||
|
| ',' { COMMA }
|
||||||
|
| '(' { LPAREN }
|
||||||
|
| ')' { RPAREN }
|
||||||
|
| '{' { LBRACE }
|
||||||
|
| '}' { RBRACE }
|
||||||
|
| '<' { LANGLE }
|
||||||
|
| '>' { RANGLE }
|
||||||
|
| '[' { LSQBRACK }
|
||||||
|
| ']' { RSQBRACK }
|
||||||
|
(* symbols *)
|
||||||
|
| '=' { EQUALS }
|
||||||
|
| '*' { STAR }
|
||||||
|
| ['x' 'X'] { X }
|
||||||
|
|
||||||
|
(* TYPES *)
|
||||||
|
| "void" { VOID }
|
||||||
|
| 'i' (pos_int as width) { INT (int_of_string width) }
|
||||||
|
| "half" { HALF }
|
||||||
|
| "float" { FLOAT }
|
||||||
|
| "double" { DOUBLE }
|
||||||
|
| "fp128" { FP128 }
|
||||||
|
| "x86_fp80" { X86_FP80 }
|
||||||
|
| "ppc_fp128" { PPC_FP128 }
|
||||||
|
| "x86_mmx" { X86_MMX }
|
||||||
|
| "label" { LABEL }
|
||||||
|
| "metadata" { METADATA }
|
||||||
|
|
||||||
|
| pos_int as size { SIZE (int_of_string size) }
|
||||||
|
(* CONSTANTS *)
|
||||||
|
| "true" { TRUE }
|
||||||
|
| "false" { FALSE }
|
||||||
|
| intlit as i { INTLIT (int_of_string i) }
|
||||||
|
(* floating point constants *)
|
||||||
|
| "null" { NULL }
|
||||||
|
|
||||||
|
(* INSTRUCTIONS *)
|
||||||
|
(* terminator instructions *)
|
||||||
|
| "ret" { RET }
|
||||||
|
| "br" { BR }
|
||||||
|
| "switch" { SWITCH }
|
||||||
|
| "indirectbr" { INDIRECTBR }
|
||||||
|
| "invoke" { INVOKE }
|
||||||
|
| "resume" { RESUME }
|
||||||
|
| "unreachable" { UNREACHABLE }
|
||||||
|
(* binary operations *)
|
||||||
|
| "add" { ADD }
|
||||||
|
| "fadd" { FADD }
|
||||||
|
| "sub" { SUB }
|
||||||
|
| "fsub" { FSUB }
|
||||||
|
| "mul" { MUL }
|
||||||
|
| "fmul" { FMUL }
|
||||||
|
| "udiv" { UDIV }
|
||||||
|
| "sdiv" { SDIV }
|
||||||
|
| "fdiv" { FDIV }
|
||||||
|
| "urem" { UREM }
|
||||||
|
| "srem" { SREM }
|
||||||
|
| "frem" { FREM }
|
||||||
|
(* arithmetic options *)
|
||||||
|
| "nuw" { NUW }
|
||||||
|
| "nsw" { NSW }
|
||||||
|
| "exact" { EXACT }
|
||||||
|
(* floating point options *)
|
||||||
|
| "nnan" { NNAN }
|
||||||
|
| "ninf" { NINF }
|
||||||
|
| "nsz" { NSZ }
|
||||||
|
| "arcp" { ARCP }
|
||||||
|
| "fast" { FAST }
|
||||||
|
(* bitwise binary operations *)
|
||||||
|
| "shl" { SHL }
|
||||||
|
| "lshr" { LSHR }
|
||||||
|
| "ashr" { ASHR }
|
||||||
|
| "and" { AND }
|
||||||
|
| "or" { OR }
|
||||||
|
| "xor" { XOR }
|
||||||
|
(* vector operations *)
|
||||||
|
| "extractelement" { EXTRACTELEMENT }
|
||||||
|
| "insertelement" { INSERTELEMENT }
|
||||||
|
| "shufflevector" { SHUFFLEVECTOR }
|
||||||
|
(* aggregate operations *)
|
||||||
|
| "extractvalue" { EXTRACTVALUE }
|
||||||
|
| "insertvalue" { INSERTVALUE }
|
||||||
|
(* memory access and addressing operations *)
|
||||||
|
| "alloca" { ALLOCA }
|
||||||
|
| "load" { LOAD }
|
||||||
|
| "store" { STORE }
|
||||||
|
| "fence" { FENCE }
|
||||||
|
| "cmpxchg" { CMPXCHG }
|
||||||
|
| "atomicrmw" { ATOMICRMW }
|
||||||
|
| "getelementptr" { GETELEMENTPTR }
|
||||||
|
(* conversion operations *)
|
||||||
|
| "trunc" { TRUNC } (* e.g. trunc ... to ... *)
|
||||||
|
| "zext" { ZEXT }
|
||||||
|
| "sext" { SEXT }
|
||||||
|
| "fptrunc" { FPTRUNC }
|
||||||
|
| "fpext" { FPEXT }
|
||||||
|
| "fptoui" { FPTOUI }
|
||||||
|
| "fptosi" { FPTOSI }
|
||||||
|
| "uitofp" { UITOFP }
|
||||||
|
| "sitofp" { SITOFP }
|
||||||
|
| "ptrtoint" { PTRTOINT }
|
||||||
|
| "inttoptr" { INTTOPTR }
|
||||||
|
| "bitcast" { BITCAST }
|
||||||
|
| "addrspacecast" { ADDRSPACECAST }
|
||||||
|
| "to" { TO } (* all conversion operations include this keyword *)
|
||||||
|
(* other operations *)
|
||||||
|
| "icmp" { ICMP }
|
||||||
|
| "fcmp" { FCMP }
|
||||||
|
| "phi" { PHI }
|
||||||
|
| "select" { SELECT }
|
||||||
|
| "call" { CALL }
|
||||||
|
| "va_arg" { VA_ARG }
|
||||||
|
| "landingpad" { LANDINGPAD }
|
||||||
|
|
||||||
|
(* identifiers - make this complete *)
|
||||||
|
| id as str { IDENT str }
|
||||||
|
|
||||||
|
| eof { EOF }
|
||||||
@ -0,0 +1,20 @@
|
|||||||
|
(*
|
||||||
|
* Copyright (c) 2015 - Facebook.
|
||||||
|
* All rights reserved.
|
||||||
|
*)
|
||||||
|
open Lexing
|
||||||
|
open Printf
|
||||||
|
|
||||||
|
exception UsageError of string
|
||||||
|
|
||||||
|
let () = try
|
||||||
|
if Array.length Sys.argv < 2 then
|
||||||
|
raise (UsageError ("Missing source file as first command line argument."))
|
||||||
|
else
|
||||||
|
let filename = Sys.argv.(1) in
|
||||||
|
let lexbuf = Lexing.from_channel (open_in filename) in
|
||||||
|
let prog = Parser.prog Lexer.token lexbuf in
|
||||||
|
let pretty = Pretty.pretty_prog prog in
|
||||||
|
print_string pretty
|
||||||
|
with
|
||||||
|
| UsageError msg -> print_string ("Usage error: " ^ msg ^ "\n")
|
||||||
@ -0,0 +1,240 @@
|
|||||||
|
(*
|
||||||
|
* Copyright (c) 2015 - Facebook.
|
||||||
|
* All rights reserved.
|
||||||
|
*)
|
||||||
|
%{
|
||||||
|
open Ast
|
||||||
|
%}
|
||||||
|
|
||||||
|
(* keywords *)
|
||||||
|
%token DEFINE
|
||||||
|
|
||||||
|
(* delimiters *)
|
||||||
|
%token COMMA
|
||||||
|
%token LPAREN
|
||||||
|
%token RPAREN
|
||||||
|
%token LBRACE
|
||||||
|
%token RBRACE
|
||||||
|
%token LANGLE
|
||||||
|
%token RANGLE
|
||||||
|
%token LSQBRACK
|
||||||
|
%token RSQBRACK
|
||||||
|
(* symbols *)
|
||||||
|
%token EQUALS
|
||||||
|
%token STAR
|
||||||
|
%token X
|
||||||
|
|
||||||
|
(* TYPES *)
|
||||||
|
%token VOID
|
||||||
|
%token <int> INT
|
||||||
|
%token HALF
|
||||||
|
%token FLOAT
|
||||||
|
%token DOUBLE
|
||||||
|
%token FP128
|
||||||
|
%token X86_FP80
|
||||||
|
%token PPC_FP128
|
||||||
|
%token X86_MMX
|
||||||
|
%token LABEL
|
||||||
|
%token METADATA
|
||||||
|
|
||||||
|
%token <int> SIZE
|
||||||
|
(* CONSTANTS *)
|
||||||
|
%token TRUE
|
||||||
|
%token FALSE
|
||||||
|
%token <int> INTLIT
|
||||||
|
%token NULL
|
||||||
|
|
||||||
|
(* INSTRUCTIONS *)
|
||||||
|
(* terminator instructions *)
|
||||||
|
%token RET
|
||||||
|
%token BR
|
||||||
|
%token SWITCH
|
||||||
|
%token INDIRECTBR
|
||||||
|
%token INVOKE
|
||||||
|
%token RESUME
|
||||||
|
%token UNREACHABLE
|
||||||
|
(* binary operations *)
|
||||||
|
%token ADD
|
||||||
|
%token FADD
|
||||||
|
%token SUB
|
||||||
|
%token FSUB
|
||||||
|
%token MUL
|
||||||
|
%token FMUL
|
||||||
|
%token UDIV
|
||||||
|
%token SDIV
|
||||||
|
%token FDIV
|
||||||
|
%token UREM
|
||||||
|
%token SREM
|
||||||
|
%token FREM
|
||||||
|
(* arithmetic options *)
|
||||||
|
%token NUW
|
||||||
|
%token NSW
|
||||||
|
%token EXACT
|
||||||
|
(* floating point options *)
|
||||||
|
%token NNAN
|
||||||
|
%token NINF
|
||||||
|
%token NSZ
|
||||||
|
%token ARCP
|
||||||
|
%token FAST
|
||||||
|
(* bitwise binary operations *)
|
||||||
|
%token SHL
|
||||||
|
%token LSHR
|
||||||
|
%token ASHR
|
||||||
|
%token AND
|
||||||
|
%token OR
|
||||||
|
%token XOR
|
||||||
|
(* vector operations *)
|
||||||
|
%token EXTRACTELEMENT
|
||||||
|
%token INSERTELEMENT
|
||||||
|
%token SHUFFLEVECTOR
|
||||||
|
(* aggregate operations *)
|
||||||
|
%token EXTRACTVALUE
|
||||||
|
%token INSERTVALUE
|
||||||
|
(* memory access and addressing operations *)
|
||||||
|
%token ALLOCA
|
||||||
|
%token LOAD
|
||||||
|
%token STORE
|
||||||
|
%token FENCE
|
||||||
|
%token CMPXCHG
|
||||||
|
%token ATOMICRMW
|
||||||
|
%token GETELEMENTPTR
|
||||||
|
(* conversion operations *)
|
||||||
|
%token TRUNC
|
||||||
|
%token ZEXT
|
||||||
|
%token SEXT
|
||||||
|
%token FPTRUNC
|
||||||
|
%token FPEXT
|
||||||
|
%token FPTOUI
|
||||||
|
%token FPTOSI
|
||||||
|
%token UITOFP
|
||||||
|
%token SITOFP
|
||||||
|
%token PTRTOINT
|
||||||
|
%token INTTOPTR
|
||||||
|
%token BITCAST
|
||||||
|
%token ADDRSPACECAST
|
||||||
|
%token TO
|
||||||
|
(* other operations *)
|
||||||
|
%token ICMP
|
||||||
|
%token FCMP
|
||||||
|
%token PHI
|
||||||
|
%token SELECT
|
||||||
|
%token CALL
|
||||||
|
%token VA_ARG
|
||||||
|
%token LANDINGPAD
|
||||||
|
|
||||||
|
%token <string> IDENT
|
||||||
|
|
||||||
|
%token EOF
|
||||||
|
|
||||||
|
%start prog
|
||||||
|
%type <Ast.prog> prog
|
||||||
|
%type <Ast.func_def> func_def
|
||||||
|
%type <Ast.typ option> ret_typ
|
||||||
|
%type <Ast.typ> typ
|
||||||
|
|
||||||
|
%%
|
||||||
|
|
||||||
|
prog:
|
||||||
|
| defs=list(func_def) EOF { Prog defs }
|
||||||
|
|
||||||
|
func_def:
|
||||||
|
| DEFINE ret_tp=ret_typ name=IDENT LPAREN params=separated_list(COMMA, pair(typ, IDENT)) RPAREN instrs=block {
|
||||||
|
FuncDef (name, ret_tp, params, instrs) }
|
||||||
|
|
||||||
|
ret_typ:
|
||||||
|
| VOID { None }
|
||||||
|
| tp=typ { Some tp }
|
||||||
|
|
||||||
|
typ:
|
||||||
|
| tp=element_typ { tp }
|
||||||
|
(*| X86_MMX { () }*)
|
||||||
|
| tp=vector_typ { tp }
|
||||||
|
| LSQBRACK sz=SIZE X tp=element_typ RSQBRACK { Tarray (sz, tp) } (* array type *)
|
||||||
|
(*| LABEL { () }
|
||||||
|
| METADATA { () }*)
|
||||||
|
(* TODO structs *)
|
||||||
|
|
||||||
|
vector_typ:
|
||||||
|
| LANGLE sz=SIZE X tp=element_typ RANGLE { Tvector (sz, tp) }
|
||||||
|
|
||||||
|
element_typ:
|
||||||
|
| width=INT { Tint width }
|
||||||
|
| floating_typ { Tfloat }
|
||||||
|
| tp=ptr_typ { tp }
|
||||||
|
|
||||||
|
floating_typ:
|
||||||
|
| HALF { () }
|
||||||
|
| FLOAT { () }
|
||||||
|
| DOUBLE { () }
|
||||||
|
| FP128 { () }
|
||||||
|
| X86_FP80 { () }
|
||||||
|
| PPC_FP128 { () }
|
||||||
|
|
||||||
|
ptr_typ:
|
||||||
|
| tp=typ STAR { Tptr tp }
|
||||||
|
|
||||||
|
block:
|
||||||
|
| LBRACE instrs=list(instr) RBRACE { instrs }
|
||||||
|
|
||||||
|
instr:
|
||||||
|
| term=terminator { term }
|
||||||
|
| IDENT EQUALS binop { Ret None }
|
||||||
|
|
||||||
|
terminator:
|
||||||
|
| RET tp=typ v=value { Ret (Some (tp, v)) }
|
||||||
|
| RET VOID { Ret None }
|
||||||
|
(*
|
||||||
|
| switch
|
||||||
|
| indirectbr
|
||||||
|
| invoke
|
||||||
|
| resume
|
||||||
|
| unreachable
|
||||||
|
*)
|
||||||
|
|
||||||
|
binop:
|
||||||
|
| ADD arith_options binop_args { () }
|
||||||
|
| FADD fast_math_flags binop_args { () }
|
||||||
|
| SUB arith_options binop_args { () }
|
||||||
|
| FSUB fast_math_flags binop_args { () }
|
||||||
|
| MUL binop_args { () }
|
||||||
|
| FMUL fast_math_flags binop_args { () }
|
||||||
|
| UDIV option(EXACT) binop_args { () }
|
||||||
|
| SDIV option(EXACT) binop_args { () }
|
||||||
|
| FDIV fast_math_flags binop_args { () }
|
||||||
|
| UREM binop_args { () }
|
||||||
|
| SREM binop_args { () }
|
||||||
|
| FREM fast_math_flags binop_args { () }
|
||||||
|
(* bitwise *)
|
||||||
|
| SHL arith_options binop_args { () }
|
||||||
|
| LSHR option(EXACT) binop_args { () }
|
||||||
|
| ASHR option(EXACT) binop_args { () }
|
||||||
|
| AND binop_args { () }
|
||||||
|
| OR binop_args { () }
|
||||||
|
| XOR binop_args { () }
|
||||||
|
(* vector *)
|
||||||
|
| EXTRACTELEMENT vector_typ value COMMA typ IDENT { () }
|
||||||
|
| INSERTELEMENT vector_typ value COMMA typ element COMMA typ IDENT { () }
|
||||||
|
|
||||||
|
arith_options:
|
||||||
|
| option(NUW) option(NSW) { () }
|
||||||
|
|
||||||
|
fast_math_flags:
|
||||||
|
| option(NNAN) option(NINF) option(NSZ) option(ARCP) option(FAST) { () }
|
||||||
|
|
||||||
|
binop_args:
|
||||||
|
| typ operand COMMA operand { () }
|
||||||
|
|
||||||
|
(* below is fuzzy *)
|
||||||
|
|
||||||
|
operand:
|
||||||
|
(* variable *)
|
||||||
|
| v=value { v }
|
||||||
|
|
||||||
|
element:
|
||||||
|
| v=value { v }
|
||||||
|
|
||||||
|
value:
|
||||||
|
| TRUE { True }
|
||||||
|
| FALSE { False }
|
||||||
|
| i=INTLIT { Intlit i }
|
||||||
|
| NULL { Null }
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
(*
|
||||||
|
* Copyright (c) 2015 - Facebook.
|
||||||
|
* All rights reserved.
|
||||||
|
*)
|
||||||
|
open Ast
|
||||||
|
|
||||||
|
let concatmap (sep : string) (f : 'a -> string) (l : 'a list) : string = String.concat sep (List.map f l)
|
||||||
|
|
||||||
|
let rec pretty_prog : prog -> string = function
|
||||||
|
Prog defs -> concatmap "" pretty_func_def defs
|
||||||
|
|
||||||
|
and pretty_func_def : func_def -> string = function
|
||||||
|
FuncDef (name, ret_tp, params, instrs) ->
|
||||||
|
"define " ^ pretty_ret_typ ret_tp ^ " " ^ name ^ "(" ^
|
||||||
|
concatmap ", " (fun (tp, id) -> pretty_typ tp ^ " " ^ id) params ^ ") {\n" ^
|
||||||
|
concatmap "" pretty_instr_ln instrs ^ "}\n"
|
||||||
|
|
||||||
|
and pretty_ret_typ : typ option -> string = function
|
||||||
|
| None -> "void"
|
||||||
|
| Some tp -> pretty_typ tp
|
||||||
|
|
||||||
|
and pretty_typ : typ -> string = function
|
||||||
|
| Tint width -> "i" ^ string_of_int width
|
||||||
|
| Tfloat (* just one type for now *) -> "float"
|
||||||
|
| Tptr tp -> pretty_typ tp ^ "*"
|
||||||
|
| Tvector (sz, tp) -> "<" ^ string_of_int sz ^ " x " ^ pretty_typ tp ^ ">"
|
||||||
|
| Tarray (sz, tp) -> "[" ^ string_of_int sz ^ " x " ^ pretty_typ tp ^ "]"
|
||||||
|
|
||||||
|
and pretty_instr_ln (i : instr) : string = pretty_instr i ^ "\n"
|
||||||
|
|
||||||
|
and pretty_instr : instr -> string = function
|
||||||
|
| Ret None -> "ret void"
|
||||||
|
| Ret (Some (tp, v)) -> "ret " ^ pretty_typ tp ^ " " ^ pretty_value v
|
||||||
|
|
||||||
|
and pretty_value : value -> string = function
|
||||||
|
| True -> "true"
|
||||||
|
| False -> "false"
|
||||||
|
| Intlit i -> string_of_int i
|
||||||
|
| Null -> "null"
|
||||||
Loading…
Reference in new issue