mirrors/codon
1
0
Fork 0
mirror of https://github.com/exaloop/codon.git synced 2025-06-03 15:03:52 +08:00
Code Issues Projects Releases Wiki Activity

Update module init codegen

Browse Source
This commit is contained in:
A. R. Shajii 2023-02-12 11:32:20 -05:00
parent 1a7a48a2fc
commit 19f91909a8
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
codon/cir/llvm/llvisitor.cpp
View File

@ -1013,6 +1013,14 @@ void LLVMVisitor::writeToPythonExtension(const PyModule &pymod,
M->getOrInsertFunction("PyModule_AddObject", i32, ptr, ptr, ptr).getCallee());
pyModuleAddObject->setDoesNotThrow();
auto *pyIncRef = llvm::cast<llvm::Function>(
M->getOrInsertFunction("Py_IncRef", B->getVoidTy(), ptr).getCallee());
pyIncRef->setDoesNotThrow();
auto *pyDecRef = llvm::cast<llvm::Function>(
M->getOrInsertFunction("Py_DecRef", B->getVoidTy(), ptr).getCallee());
pyDecRef->setDoesNotThrow();
auto *pyModuleInit = llvm::cast<llvm::Function>(
M->getOrInsertFunction("PyInit_" + pymod.name, ptr).getCallee());
auto *block = llvm::BasicBlock::Create(*context, "entry", pyModuleInit);
@ -1050,7 +1058,7 @@ void LLVMVisitor::writeToPythonExtension(const PyModule &pymod,
seqassertn(it != typeVars.end(), "type not found");
auto *typeVar = it->second;
// TODO incref typeVar
B->CreateCall(pyIncRef, typeVar);
auto *status =
B->CreateCall(pyModuleAddObject, {mod, pyString(pytype.name), typeVar});
fail = llvm::BasicBlock::Create(*context, "failure", pyModuleInit);
@ -1058,8 +1066,8 @@ void LLVMVisitor::writeToPythonExtension(const PyModule &pymod,
B->CreateCondBr(B->CreateICmpSLT(status, zero32), fail, block);
B->SetInsertPoint(fail);
// TODO decref typeVar
// TODO decref mod
B->CreateCall(pyDecRef, typeVar);
B->CreateCall(pyDecRef, mod);
B->CreateRet(null);
B->SetInsertPoint(block);