diff --git a/src/include/midend/SysYIRGenerator.h b/src/include/midend/SysYIRGenerator.h index b4d4e57..70fca7f 100644 --- a/src/include/midend/SysYIRGenerator.h +++ b/src/include/midend/SysYIRGenerator.h @@ -51,6 +51,7 @@ public: Module *pModule, IRBuilder *pBuilder); static void initExternalFunction(Module *pModule, IRBuilder *pBuilder); + static void modify_timefuncname(Module *pModule); }; class SysYIRGenerator : public SysYBaseVisitor { diff --git a/src/midend/SysYIRGenerator.cpp b/src/midend/SysYIRGenerator.cpp index b2a52be..e6840a0 100644 --- a/src/midend/SysYIRGenerator.cpp +++ b/src/midend/SysYIRGenerator.cpp @@ -674,6 +674,8 @@ std::any SysYIRGenerator::visitCompUnit(SysYParser::CompUnitContext *ctx) { pModule->enterNewScope(); visitChildren(ctx); pModule->leaveScope(); + + Utils::modify_timefuncname(pModule); return pModule; } @@ -2403,4 +2405,12 @@ void Utils::initExternalFunction(Module *pModule, IRBuilder *pBuilder) { } +void Utils::modify_timefuncname(Module *pModule){ + auto starttimeFunc = pModule->getExternalFunction("starttime"); + auto stoptimeFunc = pModule->getExternalFunction("stoptime"); + starttimeFunc->setName("_sysy_starttime"); + stoptimeFunc->setName("_sysy_stoptime"); + +} + } // namespace sysy \ No newline at end of file