作者:tomoyazhang,腾讯 PCG 后台开发工程师
这个系列来自 LLVM 的Kaleidoscope 教程,增加了我对代码的注释以及一些理解,修改了部分代码。现在开始我们要使用 LLVM 实现一个编译器,完成对如下代码的编译运行。
#斐波那契数列函数定义
deffib(x)
ifx<3then
1
else
fib(x-1)+fib(x-2)
fib(40)
#函数声明
externsin(arg)
externcos(arg)
externatan2(arg1arg2)
#声明后的函数可调用
atan2(sin(.4),cos(42))
这个语言称为 Kaleidoscope, 从代码可以看出,Kaleidoscope 支持函数、条件分支、数值计算等语言特性。为了方便,Kaleidoscope 唯一支持的数据类型为 float64, 所以示例中的所有数值都是 float64。
编译的第一个步骤称为 Lex, 词法分析,其功能是将文本输入转为多个 tokens, 比如对于如下代码:
atan2(sin(.4),cos(42))
就应该转为:
tokens=["atan2","(","sin","(",.4,")",",","cos","(",42,")",")"]
接下来我们使用 C++来写这个 Lexer, 由于这是教程代码,所以并没有使用工程项目应有的设计:
//如果不是以下5种情况,Lexer返回[0-255]的ASCII值,否则返回以下枚举值
enumToken{
TOKEN_EOF=-1,//文件结束标识符
TOKEN_DEF=-2,//关键字def
TOKEN_EXTERN=-3,//关键字extern
TOKEN_IDENTIFIER=-4,//名字
TOKEN_NUMBER=-5//数值
};
std::stringg_identifier_str;//FilledinifTOKEN_IDENTIFIER
doubleg_number_val;//FilledinifTOKEN_NUMBER
//从标准输入解析一个Token并返回
intGetToken(){
staticintlast_char='';
//忽略空白字符
while(isspace(last_char)){
last_char=getchar();
}
//识别字符串
if(isalpha(last_char)){
g_identifier_str=last_char;
while(isalnum((last_char=getchar()))){
g_identifier_str+=last_char;
}
if(g_identifier_str=="def"){
returnTOKEN_DEF;
}elseif(g_identifier_str=="extern"){
returnTOKEN_EXTERN;
}else{
returnTOKEN_IDENTIFIER;
}
}
//识别数值
if(isdigit(last_char)||last_char=='.'){
std::stringnum_str;
do{
num_str+=last_char;
last_char=getchar();
}while(isdigit(last_char)||last_char=='.');
g_number_val=strtod(num_str.c_str(),nullptr);
returnTOKEN_NUMBER;
}
//忽略注释
if(last_char=='#'){
do{
last_char=getchar();
}while(last_char!=EOF&&last_char!='\n'&&last_char!='\r');
if(last_char!=EOF){
returnGetToken();
}
}
//识别文件结束
if(last_char==EOF){
returnTOKEN_EOF;
}
//直接返回ASCII
intthis_char=last_char;
last_char=getchar();
returnthis_char;
}
使用 Lexer 对之前的代码处理结果为(使用空格分隔 tokens):
deffib(x)ifx<3then1elsefib(x-1)+fib(x-2)fib(40)externsin(arg)
externcos(arg)externatan2(arg1arg2)atan2(sin(0.4),cos(42))
Lexer 的输入是代码文本,输出是有序的一个个 Token。
编译的第二个步骤称为 Parse, 其功能是将 Lexer 输出的 tokens 转为 AST (Abstract Syntax Tree)。我们首先定义表达式的 AST Node:
//所有`表达式`节点的基类
classExprAST{
public:
virtual~ExprAST(){}
};
//字面值表达式
classNumberExprAST:publicExprAST{
public:
NumberExprAST(doubleval):val_(val){}
private:
doubleval_;
};
//变量表达式
classVariableExprAST:publicExprAST{
public:
VariableExprAST(conststd::string&name):name_(name){}
private:
std::stringname_;
};
//二元操作表达式
classBinaryExprAST:publicExprAST{
public:
BinaryExprAST(charop,std::unique_ptr<ExprAST>lhs,
std::unique_ptr<ExprAST>rhs)
:op_(op),lhs_(std::move(lhs)),rhs_(std::move(rhs)){}
private:
charop_;
std::unique_ptr<ExprAST>lhs_;
std::unique_ptr<ExprAST>rhs_;
};
//函数调用表达式
classCallExprAST:publicExprAST{
public:
CallExprAST(conststd::string&callee,
std::vector<std::unique_ptr<ExprAST>>args)
:callee_(callee),args_(std::move(args)){}
private:
std::stringcallee_;
std::vector<std::unique_ptr<ExprAST>>args_;
};
为了便于理解,关于条件表达式的内容放在后面,这里暂不考虑。接着我们定义函数声明和函数的 AST Node:
//函数接口
classPrototypeAST{
public:
PrototypeAST(conststd::string&name,std::vector<std::string>args)
:name_(name),args_(std::move(args)){}
conststd::string&name()const{returnname_;}
private:
std::stringname_;
std::vector<std::string>args_;
};
//函数
classFunctionAST{
public:
FunctionAST(std::unique_ptr<PrototypeAST>proto,
std::unique_ptr<ExprAST>body)
:proto_(std::move(proto)),body_(std::move(body)){}
private:
std::unique_ptr<PrototypeAST>proto_;
std::unique_ptr<ExprAST>body_;
};
接下来我们要进行 Parse, 在正式 Parse 前,定义如下函数方便后续处理:
intg_current_token;//当前待处理的Token
intGetNextToken(){
returng_current_token=GetToken();
}
首先我们处理最简单的字面值:
//numberexpr::=number
std::unique_ptr<ExprAST>ParseNumberExpr(){
autoresult=std::make_unique<NumberExprAST>(g_number_val);
GetNextToken();
returnstd::move(result);
}
这段程序非常简单,当前 Token 为 TOKEN_NUMBER 时被调用,使用 g_number_val,创建一个 NumberExprAST, 因为当前 Token 处理完毕,让 Lexer 前进一个 Token, 最后返回。接着我们处理圆括号操作符、变量、函数调用:
//parenexpr::=(expression)
std::unique_ptr<ExprAST>ParseParenExpr(){
GetNextToken();//eat(
autoexpr=ParseExpression();
GetNextToken();//eat)
returnexpr;
}
///identifierexpr
///::=identifier
///::=identifier(expression,expression,...,expression)
std::unique_ptr<ExprAST>ParseIdentifierExpr(){
std::stringid=g_identifier_str;
GetNextToken();
if(g_current_token!='('){
returnstd::make_unique<VariableExprAST>(id);
}else{
GetNextToken();//eat(
std::vector<std::unique_ptr<ExprAST>>args;
while(g_current_token!=')'){
args.push_back(ParseExpression());
if(g_current_token==')'){
break;
}else{
GetNextToken();//eat,
}
}
GetNextToken();//eat)
returnstd::make_unique<CallExprAST>(id,std::move(args));
}
}
上面代码中的 ParseExpression 与 ParseParenExpr 等存在循环依赖,这里按照其名字理解意思即可,具体实现在后面。我们将 NumberExpr、ParenExpr、IdentifierExpr 视为 PrimaryExpr, 封装 ParsePrimary 方便后续调用:
///primary
///::=identifierexpr
///::=numberexpr
///::=parenexpr
std::unique_ptr<ExprAST>ParsePrimary(){
switch(g_current_token){
caseTOKEN_IDENTIFIER:returnParseIdentifierExpr();
caseTOKEN_NUMBER:returnParseNumberExpr();
case'(':returnParseParenExpr();
default:returnnullptr;
}
}
接下来我们考虑如何处理二元操作符,为了方便,Kaleidoscope 只支持 4 种二元操作符,优先级为:
'<'<'+'='-'<'*'
即'<'的优先级最低,而'*'的优先级最高,在代码中实现为:
//定义优先级
conststd::map<char,int>g_binop_precedence={
{'<',10},{'+',20},{'-',20},{'*',40}};
//获得当前Token的优先级
intGetTokenPrecedence(){
autoit=g_binop_precedence.find(g_current_token);
if(it!=g_binop_precedence.end()){
returnit->second;
}else{
return-1;
}
}
对于带优先级的二元操作符的解析,我们会将其分成多个片段。比如一个表达式:
a+b+(c+d)*e*f+g
首先解析 a, 然后处理多个二元组:
[+,b],[+,(c+d)],[*,e],[*,f],[+,g]
即,复杂表达式可以抽象为一个 PrimaryExpr 跟着多个[binop, PrimaryExpr]
二元组,注意由于圆括号属于 PrimaryExpr, 所以这里不需要考虑怎么特殊处理(c+d)
,因为会被 ParsePrimary 自动处理。
//parse
//lhs[binopprimary][binopprimary]...
//如遇到优先级小于min_precedence的操作符,则停止
std::unique_ptr<ExprAST>ParseBinOpRhs(intmin_precedence,
std::unique_ptr<ExprAST>lhs){
while(true){
intcurrent_precedence=GetTokenPrecedence();
if(current_precedence<min_precedence){
//如果当前token不是二元操作符,current_precedence为-1,结束任务
//如果遇到优先级更低的操作符,也结束任务
returnlhs;
}
intbinop=g_current_token;
GetNextToken();//eatbinop
autorhs=ParsePrimary();
//现在我们有两种可能的解析方式
//*(lhsbinoprhs)binopunparsed
//*lhsbinop(rhsbinopunparsed)
intnext_precedence=GetTokenPrecedence();
if(current_precedence<next_precedence){
//将高于current_precedence的右边的操作符处理掉返回
rhs=ParseBinOpRhs(current_precedence+1,std::move(rhs));
}
lhs=
std::make_unique<BinaryExprAST>(binop,std::move(lhs),std::move(rhs));
//继续循环
}
}
//expression
//::=primary[binopprimary][binopprimary]...
std::unique_ptr<ExprAST>ParseExpression(){
autolhs=ParsePrimary();
returnParseBinOpRhs(0,std::move(lhs));
}
最复杂的部分完成后,按部就班把 function 写完:
//prototype
//::=id(idid...id)
std::unique_ptr<PrototypeAST>ParsePrototype(){
std::stringfunction_name=g_identifier_str;
GetNextToken();
std::vector<std::string>arg_names;
while(GetNextToken()==TOKEN_IDENTIFIER){
arg_names.push_back(g_identifier_str);
}
GetNextToken();//eat)
returnstd::make_unique<PrototypeAST>(function_name,std::move(arg_names));
}
//definition::=defprototypeexpression
std::unique_ptr<FunctionAST>ParseDefinition(){
GetNextToken();//eatdef
autoproto=ParsePrototype();
autoexpr=ParseExpression();
returnstd::make_unique<FunctionAST>(std::move(proto),std::move(expr));
}
//external::=externprototype
std::unique_ptr<PrototypeAST>ParseExtern(){
GetNextToken();//eatextern
returnParsePrototype();
}
最后,我们为顶层的代码实现匿名 function:
//toplevelexpr::=expression
std::unique_ptr<FunctionAST>ParseTopLevelExpr(){
autoexpr=ParseExpression();
autoproto=std::make_unique<PrototypeAST>("",std::vector<std::string>());
returnstd::make_unique<FunctionAST>(std::move(proto),std::move(expr));
}
顶层代码的意思是放在全局而不放在 function 内定义的一些执行语句比如变量赋值,函数调用等。编写一个 main 函数:
intmain(){
GetNextToken();
while(true){
switch(g_current_token){
caseTOKEN_EOF:return0;
caseTOKEN_DEF:{
ParseDefinition();
std::cout<<"parsedafunctiondefinition"<<std::endl;
break;
}
caseTOKEN_EXTERN:{
ParseExtern();
std::cout<<"parsedaextern"<<std::endl;
break;
}
default:{
ParseTopLevelExpr();
std::cout<<"parsedatoplevelexpr"<<std::endl;
break;
}
}
}
return0;
}
编译:
clang++main.cpp`llvm-config--cxxflags--ldflags--libs`
输入如下代码进行测试:
deffoo(xy)
x+foo(y,4)
deffoo(xy)
x+y
y
externsin(a)
得到输出:
parsedafunctiondefinition
parsedafunctiondefinition
parsedatoplevelexpr
parsedaextern
至此成功将 Lexer 输出的 tokens 转为 AST。
终于开始 codegen 了,首先我们 include 一些 LLVM 头文件,定义一些全局变量:
#include"llvm/ADT/APFloat.h"
#include"llvm/ADT/STLExtras.h"
#include"llvm/IR/BasicBlock.h"
#include"llvm/IR/Constants.h"
#include"llvm/IR/DerivedTypes.h"
#include"llvm/IR/Function.h"
#include"llvm/IR/IRBuilder.h"
#include"llvm/IR/LLVMContext.h"
#include"llvm/IR/LegacyPassManager.h"
#include"llvm/IR/Module.h"
#include"llvm/IR/Type.h"
#include"llvm/IR/Verifier.h"
#include"llvm/Support/TargetSelect.h"
#include"llvm/Target/TargetMachine.h"
#include"llvm/Transforms/InstCombine/InstCombine.h"
#include"llvm/Transforms/Scalar.h"
#include"llvm/Transforms/Scalar/GVN.h"
//记录了LLVM的核心数据结构,比如类型和常量表,不过我们不太需要关心它的内部
llvm::LLVMContextg_llvm_context;
//用于创建LLVM指令
llvm::IRBuilder<>g_ir_builder(g_llvm_context);
//用于管理函数和全局变量,可以粗浅地理解为类c++的编译单元(单个cpp文件)
llvm::Moduleg_module("mycooljit",g_llvm_context);
//用于记录函数的变量参数
std::map<std::string,llvm::Value*>g_named_values;
然后给每个 AST Class 增加一个 CodeGen 接口:
//所有`表达式`节点的基类
classExprAST{
public:
virtual~ExprAST(){}
virtualllvm::Value*CodeGen()=0;
};
//字面值表达式
classNumberExprAST:publicExprAST{
public:
NumberExprAST(doubleval):val_(val){}
llvm::Value*CodeGen()override;
private:
doubleval_;
};
首先实现 NumberExprAST 的 CodeGen:
llvm::Value*NumberExprAST::CodeGen(){
returnllvm::ConstantFP::get(g_llvm_context,llvm::APFloat(val_));
}
由于 Kaleidoscope 只有一种数据类型 FP64, 所以直接调用 ConstantFP 传入即可,APFloat 是 llvm 内部的数据结构,用于存储 Arbitrary Precision Float. 在 LLVM IR 中,所有常量是唯一且共享的,所以这里使用的 get 而不是 new/create。
然后实现 VariableExprAST 的 CodeGen:
llvm::Value*VariableExprAST::CodeGen(){
returng_named_values.at(name_);
}
由于 Kaleidoscope 的 VariableExpr 只存在于函数内对函数参数的引用,我们假定函数参数已经被注册到 g_name_valus 中,所以 VariableExpr 直接查表返回即可。
接着实现 BinaryExprAST, 分别 codegen lhs, rhs 然后创建指令处理 lhs, rhs 即可:
llvm::Value*BinaryExprAST::CodeGen(){
llvm::Value*lhs=lhs_->CodeGen();
llvm::Value*rhs=rhs_->CodeGen();
switch(op_){
case'<':{
llvm::Value*tmp=g_ir_builder.CreateFCmpULT(lhs,rhs,"cmptmp");
//把0/1转为0.0/1.0
returng_ir_builder.CreateUIToFP(
tmp,llvm::Type::getDoubleTy(g_llvm_context),"booltmp");
}
case'+':returng_ir_builder.CreateFAdd(lhs,rhs,"addtmp");
case'-':returng_ir_builder.CreateFSub(lhs,rhs,"subtmp");
case'*':returng_ir_builder.CreateFMul(lhs,rhs,"multmp");
default:returnnullptr;
}
}
实现 CallExprAST:
llvm::Value*CallExprAST::CodeGen(){
//g_module中存储了全局变量/函数等
llvm::Function*callee=g_module.getFunction(callee_);
std::vector<llvm::Value*>args;
for(std::unique_ptr<ExprAST>&arg_expr:args_){
args.push_back(arg_expr->CodeGen());
}
returng_ir_builder.CreateCall(callee,args,"calltmp");
}
实现 ProtoTypeAST:
llvm::Value*PrototypeAST::CodeGen(){
//创建kaleidoscope的函数类型double(doube,double,...,double)
std::vector<llvm::Type*>doubles(args_.size(),
llvm::Type::getDoubleTy(g_llvm_context));
//函数类型是唯一的,所以使用get而不是new/create
llvm::FunctionType*function_type=llvm::FunctionType::get(
llvm::Type::getDoubleTy(g_llvm_context),doubles,false);
//创建函数,ExternalLinkage意味着函数可能不在当前module中定义,在当前module
//即g_module中注册名字为name_,后面可以使用这个名字在g_module中查询
llvm::Function*func=llvm::Function::Create(
function_type,llvm::Function::ExternalLinkage,name_,&g_module);
//增加IR可读性,设置function的argumentname
intindex=0;
for(auto&arg:func->args()){
arg.setName(args_[index++]);
}
returnfunc;
}
实现 FunctionAST:
llvm::Value*FunctionAST::CodeGen(){
//检查函数声明是否已完成codegen(比如之前的extern声明),如果没有则执行codegen
llvm::Function*func=g_module.getFunction(proto_->name());
if(func==nullptr){
func=proto_->CodeGen();
}
//创建一个Block并且设置为指令插入位置。
//llvmblock用于定义controlflowgraph,由于我们暂不实现controlflow,创建
//一个单独的block即可
llvm::BasicBlock*block=
llvm::BasicBlock::Create(g_llvm_context,"entry",func);
g_ir_builder.SetInsertPoint(block);
//将函数参数注册到g_named_values中,让VariableExprAST可以codegen
g_named_values.clear();
for(llvm::Value&arg:func->args()){
g_named_values[arg.getName()]=&arg;
}
//codegenbody然后return
llvm::Value*ret_val=body_->CodeGen();
g_ir_builder.CreateRet(ret_val);
llv::verifyFunction(*func);
returnfunc;
}
至此,所有 codegen 都已完成,修改 main:
intmain(){
GetNextToken();
while(true){
switch(g_current_token){
caseTOKEN_EOF:return0;
caseTOKEN_DEF:{
autoast=ParseDefinition();
std::cout<<"parsedafunctiondefinition"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
break;
}
caseTOKEN_EXTERN:{
autoast=ParseExtern();
std::cout<<"parsedaextern"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
break;
}
default:{
autoast=ParseTopLevelExpr();
std::cout<<"parsedatoplevelexpr"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
break;
}
}
}
return0;
}
输入测试:
4+5
deffoo(ab)
a*a+2*a*b+b*b
foo(2,3)
defbar(a)
foo(a,4)+bar(31337)
externcos(x)
cos(1.234)
得到输出:
parsedatoplevelexpr
definedouble@0(){
entry:
retdouble9.000000e+00
}
parsedafunctiondefinition
definedouble@foo(double%a,double%b){
entry:
%multmp=fmuldouble%a,%a
%multmp1=fmuldouble2.000000e+00,%a
%multmp2=fmuldouble%multmp1,%b
%addtmp=fadddouble%multmp,%multmp2
%multmp3=fmuldouble%b,%b
%addtmp4=fadddouble%addtmp,%multmp3
retdouble%addtmp4
}
parsedatoplevelexpr
definedouble@1(){
entry:
%calltmp=calldouble@foo(double2.000000e+00,double3.000000e+00)
retdouble%calltmp
}
parsedafunctiondefinition
definedouble@bar(double%a){
entry:
%calltmp=calldouble@foo(double%a,double4.000000e+00)
%calltmp1=calldouble@bar(double3.133700e+04)
%addtmp=fadddouble%calltmp,%calltmp1
retdouble%addtmp
}
parsedaextern
declaredouble@cos(double)
parsedatoplevelexpr
definedouble@2(){
entry:
%calltmp=calldouble@cos(double1.234000e+00)
retdouble%calltmp
}
至此,我们已成功将 Parser 输出的 AST 转为 LLVM IR。
我们使用上一节的程序处理如下代码:
deftest(x)
1+2+x
可以得到:
parsedafunctiondefinition
definedouble@test(double%x){
entry:
%addtmp=fadddouble3.000000e+00,%x
retdouble%addtmp
}
可以看到,生成的指令直接是 1+2 的结果,而没有 1 + 2 的指令,这种自动把常量计算完毕而不是生成加法指令的优化称为 Constant Folding。
在大部分时候仅有这个优化仍然不够,比如如下代码:
deftest(x)
(1+2+x)*(x+(1+2))
可以得到编译结果:
parsedafunctiondefinition
definedouble@test(double%x){
entry:
%addtmp=fadddouble3.000000e+00,%x
%addtmp1=fadddouble%x,3.000000e+00
%multmp=fmuldouble%addtmp,%addtmp1
retdouble%multmp
}
生成了两个加法指令,但最优做法只需要一个加法即可,因为乘法的两边 lhs 和 rhs 是相等的。
这需要其他的优化技术,llvm 以"passes"的形式提供,llvm 中的 passes 可以选择是否启用,可以设置 passes 的顺序。
这里我们对每个函数单独做优化,定义 g_fpm, 增加几个 passes:
llvm::legacy::FunctionPassManagerg_fpm(&g_module);
intmain(){
g_fpm.add(llvm::createInstructionCombiningPass());
g_fpm.add(llvm::createReassociatePass());
g_fpm.add(llvm::createGVNPass());
g_fpm.add(llvm::createCFGSimplificationPass());
g_fpm.doInitialization();
...
}
在 FunctionAST 的 CodeGen 中增加一句:
llvm::Value*ret_val=body_->CodeGen();
g_ir_builder.CreateRet(ret_val);
llvm::verifyFunction(*func);
g_fpm.run(*func);//增加这句
returnfunc;
即启动了对每个 function 的优化,接下来测试之前的代码:
parsedafunctiondefinition
definedouble@test(double%x){
entry:
%addtmp=fadddouble%x,3.000000e+00
%multmp=fmuldouble%addtmp,%addtmp
retdouble%multmp
}
可以看到,和我们期望的一样,加法指令减少到一个。
由于 JIT 模式中我们需要反复创建新的 module, 所以我们将全局变量 g_module 改为 unique_ptr。
//用于管理函数和全局变量,可以粗浅地理解为类c++的编译单元(单个cpp文件)
std::unique_ptr<llvm::Module>g_module=
std::make_unique<llvm::Module>("mycooljit",g_llvm_context);
为了专注于 JIT,我们可以把优化的 passes 删掉。
修改 ParseTopLevelExpr,给 PrototypeAST 命名为__anon_expr, 让我们后面可以通过这个名字找到它。
//toplevelexpr::=expression
std::unique_ptr<FunctionAST>ParseTopLevelExpr(){
autoexpr=ParseExpression();
autoproto=
std::make_unique<PrototypeAST>("__anon_expr",std::vector<std::string>());
returnstd::make_unique<FunctionAST>(std::move(proto),std::move(expr));
}
然后我们从 llvm-project 中拷贝一份代码 llvm/examples/Kaleidoscope/include/KaleidoscopeJIT.h 到本地再 include, 其定义了 KaleidoscopeJIT 类,关于这个类,在后面会做解读,这里先不管。
定义全局变量 g_jit, 并使用 InitializeNativeTarget*函数初始化环境。
#include"KaleidoscopeJIT.h"
std::unique_ptr<llvm::orc::KaleidoscopeJIT>g_jit;
intmain(){
llvm::InitializeNativeTarget();
llvm::InitializeNativeTargetAsmPrinter();
llvm::InitializeNativeTargetAsmParser();
g_jit.reset(newllvm::orc::KaleidoscopeJIT);
g_module->setDataLayout(g_jit->getTargetMachine().createDataLayout());
...
}
修改 main 处理 top level expr 的代码为:
autoast=ParseTopLevelExpr();
std::cout<<"parsedatoplevelexpr"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cout<<std::endl;
autoh=g_jit->addModule(std::move(g_module));
//重新创建g_module在下次使用
g_module=
std::make_unique<llvm::Module>("mycooljit",g_llvm_context);
g_module->setDataLayout(g_jit->getTargetMachine().createDataLayout());
//通过名字找到编译的函数符号
autosymbol=g_jit->findSymbol("__anon_expr");
//强转为C函数指针
double(*fp)()=(double(*)())(symbol.getAddress().get());
//执行输出
std::cout<<fp()<<std::endl;
g_jit->removeModule(h);
break;
输入:
4+5
deffoo(ab)
a*a+2*a*b+b*b
foo(2,3)
得到输出:
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
retdouble9.000000e+00
}
9
parsedafunctiondefinition
definedouble@foo(double%a,double%b){
entry:
%multmp=fmuldouble%a,%a
%multmp1=fmuldouble2.000000e+00,%a
%multmp2=fmuldouble%multmp1,%b
%addtmp=fadddouble%multmp,%multmp2
%multmp3=fmuldouble%b,%b
%addtmp4=fadddouble%addtmp,%multmp3
retdouble%addtmp4
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double2.000000e+00,double3.000000e+00)
retdouble%calltmp
}
25
可以看到代码已经顺利执行,但现在的实现仍然是有问题的,比如上面的输入,foo 函数的定义和调用是被归在同一个 module 中,当第一次调用完成后,由于我们 removeModule, 第二次调用 foo 会失败。
在解决这个问题之前,我们先把 main 函数内对不同 TOKEN 的处理拆成多个函数,如下:
voidReCreateModule(){
g_module=std::make_unique<llvm::Module>("mycooljit",g_llvm_context);
g_module->setDataLayout(g_jit->getTargetMachine().createDataLayout());
}
voidParseDefinitionToken(){
autoast=ParseDefinition();
std::cout<<"parsedafunctiondefinition"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
}
voidParseExternToken(){
autoast=ParseExtern();
std::cout<<"parsedaextern"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
}
voidParseTopLevel(){
autoast=ParseTopLevelExpr();
std::cout<<"parsedatoplevelexpr"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cout<<std::endl;
autoh=g_jit->addModule(std::move(g_module));
//重新创建g_module在下次使用
ReCreateModule();
//通过名字找到编译的函数符号
autosymbol=g_jit->findSymbol("__anon_expr");
//强转为C函数指针
double(*fp)()=(double(*)())(symbol.getAddress().get());
//执行输出
std::cout<<fp()<<std::endl;
g_jit->removeModule(h);
}
intmain(){
llvm::InitializeNativeTarget();
llvm::InitializeNativeTargetAsmPrinter();
llvm::InitializeNativeTargetAsmParser();
g_jit.reset(newllvm::orc::KaleidoscopeJIT);
g_module->setDataLayout(g_jit->getTargetMachine().createDataLayout());
GetNextToken();
while(true){
switch(g_current_token){
caseTOKEN_EOF:return0;
caseTOKEN_DEF:ParseDefinitionToken();break;
caseTOKEN_EXTERN:ParseExternToken();break;
default:ParseTopLevel();break;
}
}
return0;
}
为了解决第二次调用 foo 失败的问题,我们需要让 function 和 top level expr 处于不同的 Module, 而处于不同 Module 的话,CallExprAST 的 CodeGen 在当前 module 会找不到 function, 所以需要自动在 CallExprAST 做 CodeGen 时在当前 Module 声明这个函数,即自动地增加 extern, 也就是在当前 Module 自动做对应 PrototypeAST 的 CodeGen.
首先,增加一个全局变量存储从函数名到函数接口的映射,并增加一个查询函数。
std::map<std::string,std::unique_ptr<PrototypeAST>>name2proto_ast;
llvm::Function*GetFunction(conststd::string&name){
llvm::Function*callee=g_module->getFunction(name);
if(callee!=nullptr){//当前module存在函数定义
returncallee;
}else{
//声明函数
returnname2proto_ast.at(name)->CodeGen();
}
}
更改 CallExprAST 的 CodeGen, 让其使用上面定义的 GetFuntion:
llvm::Value*CallExprAST::CodeGen(){
llvm::Functioncallee=GetFunction(callee_);
std::vector<llvm::Value*>args;
for(std::unique_ptr<ExprAST>&arg_expr:args_){
args.push_back(arg_expr->CodeGen());
}
returng_ir_builder.CreateCall(callee,args,"calltmp");
}
更改 FunctionAST 的 CodeGen, 让其将结果写入 name2proto_ast:
llvm::Value*FunctionAST::CodeGen(){
PrototypeAST&proto=*proto_;
name2proto_ast[proto.name()]=std::move(proto_);//transferownership
llvm::Function*func=GetFunction(proto.name());
//创建一个Block并且设置为指令插入位置。
//llvmblock用于定义controlflowgraph,由于我们暂不实现controlflow,创建
//一个单独的block即可
llvm::BasicBlock*block=
llvm::BasicBlock::Create(g_llvm_context,"entry",func);
g_ir_builder.SetInsertPoint(block);
//将函数参数注册到g_named_values中,让VariableExprAST可以codegen
g_named_values.clear();
for(llvm::Value&arg:func->args()){
g_named_values[arg.getName()]=&arg;
}
//codegenbody然后return
llvm::Value*ret_val=body_->CodeGen();
g_ir_builder.CreateRet(ret_val);
llvm::verifyFunction(*func);
returnfunc;
}
修改 ParseExternToken 将结果写入 name2proto_ast:
voidParseExternToken(){
autoast=ParseExtern();
std::cout<<"parsedaextern"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
name2proto_ast[ast->name()]=std::move(ast);
}
修改 ParseDefinitionToken 让其使用独立 Module:
voidParseDefinitionToken(){
autoast=ParseDefinition();
std::cout<<"parsedafunctiondefinition"<<std::endl;
ast->CodeGen()->print(llvm::errs());
std::cerr<<std::endl;
g_jit->addModule(std::move(g_module));
ReCreateModule();
}
修改完毕,输入测试:
deffoo(x)
x+1
foo(2)
deffoo(x)
x+2
foo(2)
externsin(x)
externcos(x)
sin(1.0)
deffoo(x)
sin(x)*sin(x)+cos(x)*cos(x)
foo(4)
foo(3)
得到输出:
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%addtmp=fadddouble%x,1.000000e+00
retdouble%addtmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double2.000000e+00)
retdouble%calltmp
}
3
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%addtmp=fadddouble%x,2.000000e+00
retdouble%addtmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double2.000000e+00)
retdouble%calltmp
}
4
parsedaextern
declaredouble@sin(double)
parsedaextern
declaredouble@cos(double)
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@sin(double1.000000e+00)
retdouble%calltmp
}
0.841471
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%calltmp=calldouble@sin(double%x)
%calltmp1=calldouble@sin(double%x)
%multmp=fmuldouble%calltmp,%calltmp1
%calltmp2=calldouble@cos(double%x)
%calltmp3=calldouble@cos(double%x)
%multmp4=fmuldouble%calltmp2,%calltmp3
%addtmp=fadddouble%multmp,%multmp4
retdouble%addtmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double4.000000e+00)
retdouble%calltmp
}
1
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double3.000000e+00)
retdouble%calltmp
}
1
成功运行,执行正确!代码可以正确解析 sin, cos 的原因在 KaleidoscopeJIT.h 中,截取其寻找符号的代码。
JITSymbolfindMangledSymbol(conststd::string&Name){
#ifdef_WIN32
//ThesymbollookupofObjectLinkingLayerusestheSymbolRef::SF_Exported
//flagtodecidewhetherasymbolwillbevisibleornot,whenwecall
//IRCompileLayer::findSymbolInwithExportedSymbolsOnlysettotrue.
//
//ButforWindowsCOFFobjects,thisflagiscurrentlyneverset.
//Forapotentialsolutionsee:https://reviews.llvm.org/rL258665
//Fornow,weallownon-exportedsymbolsonWindowsasaworkaround.
constboolExportedSymbolsOnly=false;
#else
constboolExportedSymbolsOnly=true;
#endif
//Searchmodulesinreverseorder:fromlastaddedtofirstadded.
//Thisistheoppositeoftheusualsearchorderfordlsym,butmakesmore
//senseinaREPLwherewewanttobindtothenewestavailabledefinition.
for(autoH:make_range(ModuleKeys.rbegin(),ModuleKeys.rend()))
if(autoSym=CompileLayer.findSymbolIn(H,Name,ExportedSymbolsOnly))
returnSym;
//Ifwecan'tfindthesymbolintheJIT,trylookinginthehostprocess.
if(autoSymAddr=RTDyldMemoryManager::getSymbolAddressInProcess(Name))
returnJITSymbol(SymAddr,JITSymbolFlags::Exported);
#ifdef_WIN32
//ForWindowsretrywithout"_"atbeginning,asRTDyldMemoryManageruses
//GetProcAddressandstandardlibrarieslikemsvcrt.dllusenames
//withandwithout"_"(forexample"_itoa"but"sin").
if(Name.length()>2&&Name[0]=='_')
if(autoSymAddr=
RTDyldMemoryManager::getSymbolAddressInProcess(Name.substr(1)))
returnJITSymbol(SymAddr,JITSymbolFlags::Exported);
#endif
returnnull
可以看到,在之前定义的 Module 找不到后会在 host process 中寻找这个符号。
继续给我们的 Kaleidoscope 添加功能之前,需要先介绍 SSA, Static Single Assignment,考虑下面代码:
y:=1
y:=2
x:=y
我们可以发现第一个赋值是不必须的,而且第三行使用的 y 来自第二行的赋值,改成 SSA 格式为
y_1=1
y_2=2
x_1=y_2
改完可以方便编译器进行优化,比如把第一个赋值删去,于是我们可以给出 SSA 的定义:
考虑如下 Control Flow Graph:
加上版本号:
可以看到,这里遇到一个问题,最下面的 block 里面的 y 应该使用 y1 还是 y2, 为了解决这个问题,插入一个特殊语句称为 phi function, 其会根据 control flow 从 y1 和 y2 中选择一个值作为 y3, 如下:
可以看到,对于 x 不需要 phi function, 因为两个分支到最后的都是 x2。
我们现在实现的 Kaleidoscope 还不够完善,缺少 if else 控制流,比如不支持如下代码:
deffib(x)
ifx<3then
1
else
fib(x-1)+fib(x-2)
首先让我们的 Lexer 能识别 if then else 三个关键字,增加 TOKEN 类型:
TOKEN_IF=-6,//if
TOKEN_THEN=-7,//then
TOKEN_ELSE=-8,//else
增加识别规则:
//识别字符串
if(isalpha(last_char)){
g_identifier_str=last_char;
while(isalnum((last_char=getchar()))){
g_identifier_str+=last_char;
}
if(g_identifier_str=="def"){
returnTOKEN_DEF;
}elseif(g_identifier_str=="extern"){
returnTOKEN_EXTERN;
}elseif(g_identifier_str=="if"){
returnTOKEN_IF;
}elseif(g_identifier_str=="then"){
returnTOKEN_THEN;
}elseif(g_identifier_str=="else"){
returnTOKEN_ELSE;
}else{
returTOKEN_IDENTIFIER;
}
}
增加 IfExprAST:
//ifthenelse
classIfExprAST:publicExprAST{
public:
IfExprAST(std::unique_ptr<ExprAST>cond,std::unique_ptr<ExprAST>then_expr,
std::unique_ptr<ExprAST>else_expr)
:cond_(std::move(cond)),
then_expr_(std::move(then_expr)),
else_expr_(std::move(else_expr)){}
llvm::Value*CodeGen()override;
private:
std::unique_ptr<ExprAST>cond_;
std::unique_ptr<ExprAST>then_expr_;
std::unique_ptr<ExprAST>else_expr_;
};
增加对 IfExprAST 的解析:
std::unique_ptr<ExprAST>ParseIfExpr(){
GetNextToken();//eatif
std::unique_ptr<ExprAST>cond=ParseExpression();
GetNextToken();//eatthen
std::unique_ptr<ExprAST>then_expr=ParseExpression();
GetNextToken();//eatelse
std::unique_ptr<ExprAST>else_expr=ParseExpression();
returnstd::make_unique<IfExprAST>(std::move(cond),std::move(then_expr),
std::move(else_expr));
}
增加到 ParsePrimary 中:
//primary
//::=identifierexpr
//::=numberexpr
//::=parenexpr
std::unique_ptr<ExprAST>ParsePrimary(){
switch(g_current_token){
caseTOKEN_IDENTIFIER:returnParseIdentifierExpr();
caseTOKEN_NUMBER:returnParseNumberExpr();
case'(':returnParseParenExpr();
caseTOKEN_IF:returnParseIfExpr();
default:returnnullptr;
}
}
完成了 lex 和 parse,接下来是最有意思的 codegen:
llvm::Value*IfExprAST::CodeGen(){
llvm::Value*cond_value=cond_->CodeGen();
//创建fcmpone指令,cond_value=(cond_value!=0.0)
//转为1bit(bool)类型
cond_value=g_ir_builder.CreateFCmpONE(
cond_value,llvm::ConstantFP::get(g_llvm_context,llvm::APFloat(0.0)),
"ifcond");
//在每个function内我们会创建一个block,这里一定在这个block内,根据block得到
//对应的上层function
llvm::Function*func=g_ir_builder.GetInsertBlock()->getParent();
//为thenelse以及最后的final创建block
llvm::BasicBlock*then_block=
llvm::BasicBlock::Create(g_llvm_context,"then",func);
llvm::BasicBlock*else_block=
llvm::BasicBlock::Create(g_llvm_context,"else");
llvm::BasicBlock*final_block=
llvm::BasicBlock::Create(g_llvm_context,"ifcont");
//创建跳转指令,根据cond_value选择then_block/else_block
g_ir_builder.CreateCondBr(cond_value,then_block,else_block);
//codegenthen_block,增加跳转final_block指令
g_ir_builder.SetInsertPoint(then_block);
llvm::Value*then_value=then_expr_->CodeGen();
g_ir_builder.CreateBr(final_block);
//then语句内可能会有嵌套的if/then/else,在嵌套的codegen时,会改变当前的
//InsertBlock,我们需要有最终结果的那个block作为这里的then_block
then_block=g_ir_builder.GetInsertBlock();
//在这里才加入是为让这个block位于上面的then里嵌套block的后面
func->getBasicBlockList().push_back(else_block);
//与then类似
g_ir_builder.SetInsertPoint(else_block);
llvm::Value*else_value=else_expr_->CodeGen();
g_ir_builder.CreateBr(final_block);
else_block=g_ir_builder.GetInsertBlock();
//codegenfinal
func->getBasicBlockList().push_back(final_block);
g_ir_builder.SetInsertPoint(final_block);
llvm::PHINode*pn=g_ir_builder.CreatePHI(
llvm::Type::getDoubleTy(g_llvm_context),2,"iftmp");
pn->addIncoming(then_value,then_block);
pn->addIncoming(else_value,else_block);
returnpn;
}
这里使用了上一节 SSA 中提到的 phi function,输入:
deffoo(x)
ifx<3then
1
else
foo(x-1)+foo(x-2)
foo(1)
foo(2)
foo(3)
foo(4)
得到输出:
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%cmptmp=fcmpultdouble%x,3.000000e+00
%booltmp=uitofpi1%cmptmptodouble
%ifcond=fcmponedouble%booltmp,0.000000e+00
bri1%ifcond,label%then,label%else
then:;preds=%entry
brlabel%ifcont
else:;preds=%entry
%subtmp=fsubdouble%x,1.000000e+00
%calltmp=calldouble@foo(double%subtmp)
%subtmp1=fsubdouble%x,2.000000e+00
%calltmp2=calldouble@foo(double%subtmp1)
%addtmp=fadddouble%calltmp,%calltmp2
brlabel%ifcont
ifcont:;preds=%else,%then
%iftmp=phidouble[1.000000e+00,%then],[%addtmp,%else]
retdouble%iftmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double1.000000e+00)
retdouble%calltmp
}
1
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double2.000000e+00)
retdouble%calltmp
}
1
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double3.000000e+00)
retdouble%calltmp
}
2
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@foo(double4.000000e+00)
retdouble%calltmp
}
3
成功完成了斐波那契数列的计算,接下来我们需要增加循环的支持,在此之前我们实现一个 printd 函数:
extern"C"doubleprintd(doublex){
printf("%lf\n",x);
return0.0;
}
编译:
clang++-gmain.cpp\`llvm-config--cxxflags--ldflags--libs\`-Wl,-no-as-needed-rdynamic
输入:
externprintd(x)
printd(12)
得到输出:
parsedaextern
declaredouble@printd(double)
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%calltmp=calldouble@printd(double1.200000e+01)
retdouble%calltmp
}
12.000000
0
可以看到,我们成功给 Kaleiscope 添加了 printd 函数,接下来看我们需要实现的循环语法, 使用 C++代码作为注释:
defprintstar(n):
fori=1,i<n,1.0in#for(doublei=1.0;i<n;i+=1.0)
printd(n)
同样,我们增加 for 和 in 的 TOKEN:
enumToken{
TOKEN_EOF=-1,//文件结束标识符
TOKEN_DEF=-2,//关键字def
TOKEN_EXTERN=-3,//关键字extern
TOKEN_IDENTIFIER=-4,//名字
TOKEN_NUMBER=-5,//数值
TOKEN_IF=-6,//if
TOKEN_THEN=-7,//then
TOKEN_ELSE=-8,//else
TOKEN_FOR=-9,//for
TOKEN_IN=-10//in
};
增加 TOKEN 的识别:
//识别字符串
if(isalpha(last_char)){
g_identifier_str=last_char;
while(isalnum((last_char=getchar()))){
g_identifier_str+=last_char;
}
if(g_identifier_str=="def"){
returnTOKEN_DEF;
}elseif(g_identifier_str=="extern"){
returnTOKEN_EXTERN;
}elseif(g_identifier_str=="if"){
returnTOKEN_IF;
}elseif(g_identifier_str=="then"){
returnTOKEN_THEN;
}elseif(g_identifier_str=="else"){
returnTOKEN_ELSE;
}elseif(g_identifier_str=="for"){
returnTOKEN_FOR;
}elseif(g_identifier_str=="in"){
returnTOKEN_IN;
}else{
returnTOKEN_IDENTIFIER;
}
}
增加 ForExprAST:
//forin
classForExprAST:publicExprAST{
public:
ForExprAST(conststd::string&var_name,std::unique_ptr<ExprAST>start_expr,
std::unique_ptr<ExprAST>end_expr,
std::unique_ptr<ExprAST>step_expr,
std::unique_ptr<ExprAST>body_expr)
:var_name_(var_name),
start_expr_(std::move(start_expr)),
end_expr_(std::move(end_expr)),
step_expr_(std::move(step_expr)),
body_expr_(std::move(body_expr)){}
llvm::Value*CodeGen()override;
private:
std::stringvar_name_;
std::unique_ptr<ExprAST>start_expr_;
std::unique_ptr<ExprAST>end_expr_;
std::unique_ptr<ExprAST>step_expr_;
std::unique_ptr<ExprAST>body_expr_;
};
添加到 Primary 的解析中:
//forexpr::=forvar_name=start_expr,end_expr,step_exprinbody_expr
std::unique_ptr<ExprAST>ParseForExpr(){
GetNextToken();//eatfor
std::stringvar_name=g_identifier_str;
GetNextToken();//eatvar_name
GetNextToken();//eat=
std::unique_ptr<ExprAST>start_expr=ParseExpression();
GetNextToken();//eat,
std::unique_ptr<ExprAST>end_expr=ParseExpression();
GetNextToken();//eat,
std::unique_ptr<ExprAST>step_expr=ParseExpression();
GetNextToken();//eatin
std::unique_ptr<ExprAST>body_expr=ParseExpression();
returnstd::make_unique<ForExprAST>(var_name,std::move(start_expr),
std::move(end_expr),std::move(step_expr),
std::move(body_expr));
}
//primary
//::=identifierexpr
//::=numberexpr
//::=parenexpr
std::unique_ptr<ExprAST>ParsePrimary(){
switch(g_current_token){
caseTOKEN_IDENTIFIER:returnParseIdentifierExpr();
caseTOKEN_NUMBER:returnParseNumberExpr();
case'(':returnParseParenExpr();
caseTOKEN_IF:returnParseIfExpr();
caseTOKEN_FOR:returnParseForExpr();
default:returnnullptr;
}
}
开始 codegen:
llvm::Value*ForExprAST::CodeGen(){
//codegenstart
llvm::Value*start_val=start_expr_->CodeGen();
//获取当前function
llvm::Function*func=g_ir_builder.GetInsertBlock()->getParent();
//保存当前的block
llvm::BasicBlock*pre_block=g_ir_builder.GetInsertBlock();
//新增一个loopblock到当前function
llvm::BasicBlock*loop_block=
llvm::BasicBlock::Create(g_llvm_context,"loop",func);
//为当前block增加到loop_block的跳转指令
g_ir_builder.CreateBr(loop_block);
//开始在loop_block内增加指令
g_ir_builder.SetInsertPoint(loop_block);
llvm::PHINode*var=g_ir_builder.CreatePHI(
llvm::Type::getDoubleTy(g_llvm_context),2,var_name_.c_str());
//如果来自pre_block的跳转,则取start_val的值
var->addIncoming(start_val,pre_block);
//现在我们新增了一个变量var,因为可能会被后面的代码引用,所以要注册到
//g_named_values中,其可能会和函数参数重名,但我们这里为了方便不管
//这个特殊情况,直接注册到g_named_values中,
g_named_values[var_name_]=var;
//在loop_block中增加body的指令
body_expr_->CodeGen();
//codegenstep_expr
llvm::Value*step_value=step_expr_->CodeGen();
//next_var=var+step_value
llvm::Value*next_value=g_ir_builder.CreateFAdd(var,step_value,"nextvar");
//codegenend_expr
llvm::Value*end_value=end_expr_->CodeGen();
//end_value=(end_value!=0.0)
end_value=g_ir_builder.CreateFCmpONE(
end_value,llvm::ConstantFP::get(g_llvm_context,llvm::APFloat(0.0)),
"loopcond");
//和if/then/else一样,这里的block可能会发生变化,保存当前的block
llvm::BasicBlock*loop_end_block=g_ir_builder.GetInsertBlock();
//创建循环结束后的block
llvm::BasicBlock*after_block=
llvm::BasicBlock::Create(g_llvm_context,"afterloop",func);
//根据end_value选择是再来一次loop_block还是进入after_block
g_ir_builder.CreateCondBr(end_value,loop_block,after_block);
//给after_block增加指令
g_ir_builder.SetInsertPoint(after_block);
//如果是再次循环,取新的值
var->addIncoming(next_value,loop_end_block);
//循环结束,避免被再次引用
g_named_values.erase(var_name_);
//return0
llvm::Constant::getNullValue(llvm::Type::getDoubleTy(g_llvm_context));
}
输入:
externprintd(x)
deffoo(x)
ifx<3then
1
else
foo(x-1)+foo(x-2)
fori=1,i<10,1.0in
printd(foo(i))
输出:
parsedaextern
declaredouble@printd(double)
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%cmptmp=fcmpultdouble%x,3.000000e+00
%booltmp=uitofpi1%cmptmptodouble
%ifcond=fcmponedouble%booltmp,0.000000e+00
bri1%ifcond,label%then,label%else
then:;preds=%entry
brlabel%ifcont
else:;preds=%entry
%subtmp=fsubdouble%x,1.000000e+00
%calltmp=calldouble@foo(double%subtmp)
%subtmp1=fsubdouble%x,2.000000e+00
%calltmp2=calldouble@foo(double%subtmp1)
%addtmp=fadddouble%calltmp,%calltmp2
brlabel%ifcont
ifcont:;preds=%else,%then
%iftmp=phidouble[1.000000e+00,%then],[%addtmp,%else]
retdouble%iftmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
brlabel%loop
loop:;preds=%loop,%entry
%i=phidouble[1.000000e+00,%entry],[%nextvar,%loop]
%calltmp=calldouble@foo(double%i)
%calltmp1=calldouble@printd(double%calltmp)
%nextvar=fadddouble%i,1.000000e+00
%cmptmp=fcmpultdouble%i,1.000000e+01
%booltmp=uitofpi1%cmptmptodouble
%loopcond=fcmponedouble%booltmp,0.000000e+00
bri1%loopcond,label%loop,label%afterloop
afterloop:;preds=%loop
retdouble0.000000e+00
}
1.000000
1.000000
2.000000
3.000000
5.000000
8.000000
13.000000
21.000000
34.000000
55.000000
0
成功遍历了斐波那契数列。
在 C++中,用户可以重载操作符而不能增加操作符。在这里,我们将给 Kaleidoscope 增加一个功能,让用户可以增加二元操作符。
#新增二元操作符`>`,优先级等于内置的`<`
defbinary>10(LHSRHS)
RHS<LHS
#新增二元操作符`|`,优先级为5
defbinary|5(LHSRHS)
ifLHSthen
1
elseifRHSthen
1
else
0
#新增二元操作符`=`,优先级为9,这个操作符类似C++的`==`
defbinary=9(LHSRHS)
!(LHS<RHS|LHS>RHS)
增加 TOKEN 的类型:
enumToken{
...
TOKEN_BINARY=-11,//binary
};
增加 TOKEN 的识别:
//从标准输入解析一个Token并返回
intGetToken(){
...
//识别字符串
if(isalpha(last_char)){
...
if(g_identifier_str=="def"){
returnTOKEN_DEF;
}elseif(g_identifier_str=="extern"){
returnTOKEN_EXTERN;
}elseif(g_identifier_str=="if"){
returnTOKEN_IF;
}elseif(g_identifier_str=="then"){
returnTOKEN_THEN;
}elseif(g_identifier_str=="else"){
returnTOKEN_ELSE;
}elseif(g_identifier_str=="for"){
returnTOKEN_FOR;
}elseif(g_identifier_str=="in"){
returnTOKEN_IN;
}elseif(g_identifier_str=="binary"){
returnTOKEN_BINARY;
}else{
returnTOKEN_IDENTIFIER;
}
}
...
}
我们把新增的二元操作符视为一个函数,所以不需要新增 AST,但是需要修改 PrototypeAST。
//函数接口
classPrototypeAST{
public:
PrototypeAST(conststd::string&name,std::vector<std::string>args,
boolis_operator=false,intop_precedence=0)
:name_(name),
args_(std::move(args)),
is_operator_(is_operator),
op_precedence_(op_precedence){}
llvm::Function*CodeGen();
conststd::string&name()const{returnname_;}
intop_precedence()const{returnop_precedence_;}
boolIsUnaryOp()const{returnis_operator_&&args_.size()==1;}
boolIsBinaryOp()const{returnis_operator_&&args_.size()==2;}
//like`|`in`binary|`
charGetOpName(){returnname_[name_.size()-1];}
private:
std::stringname_;
std::vector<std::string>args_;
boolis_operator_;
intop_precedence_;
};
修改 parse 部分:
//prototype
//::=id(idid...id)
//::=binarybinopprecedence(idid)
std::unique_ptr<PrototypeAST>ParsePrototype(){
std::stringfunction_name;
boolis_operator=false;
intprecedence=0;
switch(g_current_token){
caseTOKEN_IDENTIFIER:{
function_name=g_identifier_str;
is_operator=false;
GetNextToken();//eatid
break;
}
caseTOKEN_BINARY:{
GetNextToken();//eatbinary
function_name="binary";
function_name+=(char)(g_current_token);
is_operator=true;
GetNextToken();//eatbinop
precedence=g_number_val;
GetNextToken();//eatprecedence
break;
}
}
std::vector<std::string>arg_names;
while(GetNextToken()==TOKEN_IDENTIFIER){
arg_names.push_back(g_identifier_str);
}
GetNextToken();//eat)
returnstd::make_unique<PrototypeAST>(function_name,arg_names,is_operator,
precedence);
}
修改 BinaryExprAST 的 CodeGen 处理自定义 Operator, 增加函数调用指令:
llvm::Value*BinaryExprAST::CodeGen(){
llvm::Value*lhs=lhs_->CodeGen();
llvm::Value*rhs=rhs_->CodeGen();
switch(op_){
case'<':{
llvm::Value*tmp=g_ir_builder.CreateFCmpULT(lhs,rhs,"cmptmp");
//把0/1转为0.0/1.0
returng_ir_builder.CreateUIToFP(
tmp,llvm::Type::getDoubleTy(g_llvm_context),"booltmp");
}
case'+':returng_ir_builder.CreateFAdd(lhs,rhs,"addtmp");
case'-':returng_ir_builder.CreateFSub(lhs,rhs,"subtmp");
case'*':returng_ir_builder.CreateFMul(lhs,rhs,"multmp");
default:{
//userdefinedoperator
llvm::Function*func=GetFunction(std::string("binary")+op_);
llvm::Value*operands[2]={lhs,rhs};
returng_ir_builder.CreateCall(func,operands,"binop");
}
}
}
在 FunctionAST 的 CodeGen 时,注册操作符优先级,从而让自定义操作符被识别为操作符。
llvm::Value*FunctionAST::CodeGen(){
PrototypeAST&proto=*proto_;
name2proto_ast[proto.name()]=std::move(proto_);//transferownership
llvm::Function*func=GetFunction(proto.name());
if(proto.IsBinaryOp()){
g_binop_precedence[proto.GetOpName()]=proto.op_precedence();
}
//创建一个Block并且设置为指令插入位置。
//llvmblock用于定义controlflowgraph,由于我们暂不实现controlflow,创建
//一个单独的block即可
llvm::BasicBlock*block=
llvm::BasicBlock::Create(g_llvm_context,"entry",func);
g_ir_builder.SetInsertPoint(block);
//将函数参数注册到g_named_values中,让VariableExprAST可以codegen
g_named_values.clear();
for(llvm::Value&arg:func->args()){
g_named_values[arg.getName()]=&arg;
}
//codegenbody然后return
llvm::Value*ret_val=body_->CodeGen();
g_ir_builder.CreateRet(ret_val);
llvm::verifyFunction(*func);
returnfunc;
}
输入:
#新增二元操作符`>`,优先级等于内置的`<`
defbinary>10(LHSRHS)
RHS<LHS
1>2
2>1
#新增二元操作符`|`,优先级为5
defbinary|5(LHSRHS)
ifLHSthen
1
elseifRHSthen
1
else
0
1|0
0|1
0|0
1|1
得到输出:
parsedafunctiondefinition
definedouble@"binary>"(doube%LHS,double%RHS){
entry:
%cmptmp=fcmpultdouble%RHS,%LHS
%booltmp=uitofpi1%cmptmptodouble
retdouble%booltmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary>"(double1.000000e+00,double2.000000e+00)
retdouble%binop
}
0
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary>"(double2.000000e+00,double1.000000e+00)
retdouble%binop
}
1
parsedafunctiondefinition
definedouble@"binary|"(double%LHS,double%RHS){
entry:
%ifcond=fcmponedouble%LHS,0.000000e+00
bri1%ifcond,label%then,label%else
then:;preds=%entry
brlabel%ifcont4
else:;preds=%entry
%ifcond1=fcmponedouble%RHS,0.000000e+00
bri1%ifcond1,label%then2,label%else3
then2:;preds=%else
brlabel%ifcont
else3:;preds=%else
brlabel%ifcont
ifcont:;preds=%else3,%then2
%iftmp=phidouble[1.000000e+00,%then2],[0.000000e+00,%else3]
brlabel%ifcont4
ifcont4:;preds=%ifcont,%then
%iftmp5=phidouble[1.000000e+00,%then],[%iftmp,%ifcont]
retdouble%iftmp5
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary|"(double1.000000e+00,double0.000000e+00)
retdouble%binop
}
1
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary|"(double0.000000e+00,double1.000000e+00)
retdouble%binop
}
1
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary|"(double0.000000e+00,double0.000000e+00)
retdouble%binop
}
0
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%binop=calldouble@"binary|"(double1.000000e+00,double1.000000e+00)
retdouble%binop
}
1
本节我们将让 Kaleidoscope 支持可变变量,首先我们看如下 C 代码:
intG,H;
inttest(_BoolCondition){
intX;
if(Condition)
X=G;
else
X=H;
returnX;
}
由于变量 X 的值依赖于程序的执行路径,会加入一个 phi node 来选取分支结果。上面代码的 LLVM IR 如下:
@G=weakglobali320;typeof@Gisi32*
@H=weakglobali320;typeof@Hisi32*
definei32@test(i1%Condition){
entry:
bri1%Condition,label%cond_true,label%cond_false
cond_true:
%X.0=loadi32*@G
brlabel%cond_next
cond_false:
%X.1=loadi32*@H
brlabel%cond_next
cond_next:
%X.2=phii32[%X.1,%cond_false],[%X.0,%cond_true]
reti32%X.2
}
上面的 X 是符合 SSA 格式的,但是这里真正的难题是给可变变量赋值时怎么自动添加 phi node。我们先了解一些信息,LLVM 要求寄存器变量是 SSA 格式,但却不允许内存对象是 SSA 格式。比如上面的例子中,G 和 H 就没有版本号。在 LLVM 中,所有内存访问都是显示的 load/store 指令,并且不存在取内存地址的操作。注意上面的例子中,即使@G/@H 全局变量定义时用的 i32, 但其类型仍然是 i32*, 表示在全局数据区存放 i32 的空间地址。
现在假设我们想创建一个类似@G 但是在栈上的内存变量,基本指令如下:
definei32@example(){entry:
%X=allocai32;typeof%Xisi32*.
...
%tmp=loadi32*%X;loadthestackvalue%Xfromthestack.
%tmp2=addi32%tmp,1;incrementit
storei32%tmp2,i32*%X;storeitback
...
于是我们可以把上面使用 phi node 的 LLVM IR 改写为使用栈上变量:
@G=weakglobali320;typeof@Gisi32*
@H=weakglobali320;typeof@Hisi32*
definei32@test(i1%Condition){
entry:
%X=allocai32;typeof%Xisi32*.
bri1%Condition,label%cond_true,label%cond_false
cond_true:
%X.0=loadi32*@G
storei32%X.0,i32*%X;UpdateX
brlabel%cond_next
cond_false:
%X.1=loadi32*@H
storei32%X.1,i32*%X;UpdateX
brlabel%cond_next
cond_next:
%X.2=loadi32*%X;ReadX
reti32%X.2
}
于是我们找到了一个处理任意可变变量而且不需要创建 phi node 的办法:
但是这会带来一个新的问题,因为内存速度不如寄存器,大量使用栈会有性能问题。不过,LLVM 优化器有一个 pass 称为"mem2reg", 专门将 stack 的使用自动地尽可能转为使用 phi node, 下面为自动优化的结果:
@G=weakglobali320
@H=weakglobali320
definei32@test(i1%Condition){
entry:
bri1%Condition,label%cond_true,label%cond_false
cond_true:
%X.0=loadi32*@G
brlabel%cond_next
cond_false:
%X.1=loadi32*@H
brlabel%cond_next
cond_next:
%X.01=phii32[%X.1,%cond_false],[%X.0,%cond_true]
reti32%X.01}
mem2reg 实现了一个称为"iterated dominance frontier"的标准算法来自动创建 SSA 格式。对 mem2reg 的使用需要注意:
因为我们后面需要启用 mem2reg,我们先把优化器加回来,修改全局定义:
std::unique_ptr<llvm::Module>g_module;
std::unique_ptr<llvm::legacy::FunctionPassManager>g_fpm;
修改 ReCreateModule:
voidReCreateModule(){
g_module=std::make_unique<llvm::Module>("mycooljit",g_llvm_context);
g_module->setDataLayout(g_jit->getTargetMachine().createDataLayout());
g_fpm=std::make_unique<llvm::legacy::FunctionPassManager>(g_module.get());
g_fpm->add(llvm::createInstructionCombiningPass));
g_fpm->add(llvm::createReassociatePass());
g_fpm->add(llvm::createGVNPass());
g_fpm->add(llvm::createCFGSimplificationPass());
g_fpm->doInitialization();
}
在 FunctionAST::CodeGen 中执行优化器:
g_ir_builder.CreateRet(ret_val);
llvm::verifyFunction(*func);
g_fpm->run(*func);
修改 main:
intmain(){
llvm::InitializeNativeTarget();
llvm::InitializeNativeTargetAsmPrinter();
llvm::InitializeNativeTargetAsmParser();
g_jit.reset(newllvm::orc::KaleidoscopeJIT);
ReCreateModule();
...
}
我们有两种类型的变量,分别是函数参数以及 for 循环的变量,这里我们将这两种变量也修改为使用内存,再让 mem2reg 进行优化。因为所有的变量都会使用内存,修改 g_named_value 存储的类型为 AllocaInst*:
std::map<std::string,llvm::AllocaInst*>g_named_values;
编写一个函数 CreateEntryBlockAlloca,简化后续工作,其功能是往函数的 EntryBlock 的最开始的地方添加分配内存指令:
llvm::AllocaInst*CreateEntryBlockAlloca(llvm::Function*func,
conststd::string&var_name){
llvm::IRBuilder<>ir_builder(&(func->getEntryBlock()),
func->getEntryBlock().begin());
returnir_builder.CreateAlloca(llvm::Type::getDoubleTy(g_llvm_context),0,
var_name.c_str());
}
修改 VariableExprAST::CodeGen, 由于我们所有变量都放在内存你上,所以增加 load 指令:
llvm::Value*VariableExprAST::CodeGen(){
llvm::AllocaInst*val=g_named_values.at(name_);
returng_ir_builder.CreateLoad(val,name_.c_str());
}
接下来我们修改 for 循环里变量的 CodeGen:
llvm::Value*ForExprAST::CodeGen(){
//获取当前function
llvm::Function*func=g_ir_builder.GetInsertBlock()->getParent();
//将变量创建为栈上变量,不再是phinode
llvm::AllocaInst*var=CreateEntryBlockAlloca(func,var_name_);
//codegenstart
llvm::Value*start_val=start_expr_->CodeGen();
//将初始值赋给var
g_ir_builder.CreateStore(start_val,var);
//新增一个loopblock到当前function
llvm::BasicBlock*loop_block=
llvm::BasicBlock::Create(g_llvm_context,"loop",func);
//为当前block增加到loop_block的跳转指令
g_ir_builder.CreateBr(loop_block);
//开始在loop_block内增加指令
g_ir_builder.SetInsertPoint(loop_block);
//现在我们新增了一个变量var,因为可能会被后面的代码引用,所以要注册到
//g_named_values中,其可能会和函数参数重名,但我们这里为了方便不管
//这个特殊情况,直接注册到g_named_values中,
g_named_values[var_name_]=var;
//在loop_block中增加body的指令
body_expr_->CodeGen();
//codegenstep_expr
llvm::Value*step_value=step_expr_->CodeGen();
//var=var+step_value
llvm::Value*cur_value=g_ir_builder.CreateLoad(var);
llvm::Value*next_value=
g_ir_builder.CreateFAdd(cur_value,step_value,"nextvar");
g_ir_builder.CreateStore(next_value,var);
//codegenend_expr
llvm::Value*end_value=end_expr_->CodeGen();
//end_value=(end_value!=0.0)
end_value=g_ir_builder.CreateFCmpONE(
end_value,llvm::ConstantFP::get(g_llvm_context,llvm::APFloat(0.0)),
"loopcond");
//和if/then/else一样,这里的block可能会发生变化,保存当前的block
llvm::BasicBlock*loop_end_block=g_ir_builder.GetInsertBlock();
//创建循环结束后的block
llvm::BasicBlock*after_block=
llvm::BasicBlock::Create(g_llvm_context,"afterloop",func);
//根据end_value选择是再来一次loop_block还是进入after_block
g_ir_builder.CreateCondBr(end_value,loop_block,after_block);
//给after_block增加指令
g_ir_builder.SetInsertPoint(after_block);
//循环结束,避免被再次引用
g_named_values.erase(var_name_);
//return0
returnllvm::Constant::getNullValue(llvm::Type::getDoubleTy(g_llvm_context));
}
修改 FunctionAST::codegen()使得参数可变:
llvm::Value*FunctionAST::CodeGen(){
PrototypeAST&proto=*proto_;
name2proto_ast[proto.name()]=std::move(proto_);//transferownership
llvm::Function*func=GetFunction(proto.name());
if(proto.IsBinaryOp()){
g_binop_precedence[proto.GetOpName()]=proto.op_precedence();
}
//创建一个Block并且设置为指令插入位置。
//llvmblock用于定义controlflowgraph,由于我们暂不实现controlflow,创建
//一个单独的block即可
llvm::BasicBlock*block=
llvm::BasicBlock::Create(g_llvm_context,"entry",func);
g_ir_builder.SetInsertPoint(block);
//将函数参数注册到g_named_values中,让VariableExprAST可以codegen
g_named_values.clear();
for(llvm::Value&arg:func->args()){
//为每个参数创建一个栈上变量,并赋初值,修改g_named_values使得后面的引用
//会引用这个栈上变量
llvm::AllocaInst*var=CreateEntryBlockAlloca(func,arg.getName());
g_ir_builder.CreateStore(&arg,var);
g_named_values[arg.getName()]=var;
}
//codegenbody然后return
llvm::Value*ret_val=body_->CodeGen();
g_ir_builder.CreateRet(ret_val);
llvm::verifyFunction(*func);
g_fpm->run(*func);
returnfunc;
}
输入:
externprintd(x)
deffoo(x)
ifx<3then
1
else
foo(x-1)+foo(x-2)
fori=1,i<10,1.0in
printd(foo(i))
输出:
parsedaextern[13/48988]
declaredouble@printd(double)
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%x1=allocadouble,align8
storedouble%x,double*%x1,align8
%cmptmp=fcmpultdouble%x,3.000000e+00
bri1%cmptmp,label%ifcont,label%else
else:;preds=%entry
%subtmp=fadddouble%x,-1.000000e+00
%calltmp=calldouble@foo(double%subtmp)
%subtmp5=fadddouble%x,-2.000000e+00
%calltmp6=calldouble@foo(double%subtmp5)
%addtmp=fadddouble%calltmp,%calltmp6
brlabel%ifcont
ifcont:;preds=%entry,%else
%iftmp=phidouble[%addtmp,%else],[1.000000e+00,%entry]
retdouble%iftmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
%i=allocadouble,align8
storedouble1.000000e+00,double*%i,align8
brlabel%loop
loop:;preds=%loop,%entry
%i1=phidouble[%nextvar,%loop],[1.000000e+00,%entry]
%calltmp=calldouble@foo(double%i1)
%calltmp2=calldouble@printd(double%calltmp)
%nextvar=fadddouble%i1,1.000000e+00
storedouble%nextvar,double*%i,align8
%cmptmp=fcmpultdouble%nextvar,1.000000e+01
bri1%cmptmp,label%loop,label%afterloop
afterloop:;preds=%loop
retdouble0.000000e+00
}
1.000000
1.000000
2.000000
3.000000
5.000000
8.000000
13.000000
21.000000
34.000000
0
可以看到,新版本的 IR 中已经没有了 phi node, 接下来我们加入优化器:
g_fpm->add(llvm::createPromoteMemoryToRegisterPass());
g_fpm->add(llvm::createInstructionCombiningPass());
g_fpm->add(llvm::createReassociatePass());
再次得到输出:
parsedaextern
declaredouble@printd(double)
parsedafunctiondefinition
definedouble@foo(double%x){
entry:
%cmptmp=fcmpultdouble%x,3.000000e+00
bri1%cmptmp,label%ifcont,label%else
else:;preds=%entry
%subtmp=fadddouble%x,-1.000000e+00
%calltmp=calldouble@foo(double%subtmp)
%subtmp5=fadddouble%x,-2.000000e+00
%calltmp6=calldouble@foo(double%subtmp5)
%addtmp=fadddouble%calltmp,%calltmp6
brlabel%ifcont
ifcont:;preds=%entry,%else
%iftmp=phidouble[%addtmp,%else],[1.000000e+00,%entry]
retdouble%iftmp
}
parsedatoplevelexpr
definedouble@__anon_expr(){
entry:
brlabel%loop
loop:;preds=%loop,%entry
%i1=phidouble[%nextvar,%loop],[1.000000e+00,%entry]
%calltmp=calldouble@foo(double%i1)
%calltmp2=calldouble@printd(double%calltmp)
%nextvar=fadddouble%i1,1.000000e+00
%cmptmp=fcmpultdouble%nextvar,1.000000e+01
bri1%cmptmp,label%loop,label%afterloop
afterloop:;preds=%loop
retdouble0.000000e+00
}
1.000000
1.000000
2.000000
3.000000
5.000000
8.000000
13.000000
21.000000
34.000000
0
可以看到,栈上变量自动地变为寄存器变量,且 phi node 自动地被添加。
完整代码见:
https://zhuanlan.zhihu.com/p/336929719
参考:
欢迎大家多多交流,共同进步。
文章转发自腾讯技术工程微信公众号,版权归其所有。文章内容不代表本站立场和任何投资暗示。
Copyright © 2021.Company 元宇宙YITB.COM All rights reserved.元宇宙YITB.COM