def _emit_file(self, module, cgft):
pm = api.llvm.PassManager.new()
os = extra.make_raw_ostream_for_printing()
pm.add(api.llvm.DataLayout.new(str(self.target_data)))
failed = self._ptr.addPassesToEmitFile(pm, os, cgft)
pm.run(module)
CGFT = api.llvm.TargetMachine.CodeGenFileType
if cgft == CGFT.CGFT_ObjectFile:
return os.bytes()
else:
return os.str()
def _emit_file(self, module, cgft):
pm = api.llvm.PassManager.new()
os = extra.make_raw_ostream_for_printing()
pm.add(api.llvm.DataLayout.new(str(self.target_data)))
failed = self._ptr.addPassesToEmitFile(pm, os, cgft)
pm.run(module)
CGFT = api.llvm.TargetMachine.CodeGenFileType
if cgft == CGFT.CGFT_ObjectFile:
return os.bytes()
else:
return os.str()
def __str__(self):
from llvmpy import extra
os = extra.make_raw_ostream_for_printing()
self.print_(os, None)
return os.str()
def test_engine_builder():
llvm.InitializeNativeTarget()
context = llvm.getGlobalContext()
m = llvm.Module.new("modname", context)
int32ty = llvm.Type.getIntNTy(context, 32)
fnty = llvm.FunctionType.get(int32ty, [int32ty], False)
fn = m.getOrInsertFunction("foo", fnty)._downcast(llvm.Function)
bb = llvm.BasicBlock.Create(context, "entry", fn, None)
builder = llvm.IRBuilder.new(context)
builder.SetInsertPoint(bb)
builder.CreateRet(fn.getArgumentList()[0])
print(fn)
eb = llvm.EngineBuilder.new(m)
eb2 = eb.setEngineKind(llvm.EngineKind.Kind.JIT)
assert eb is eb2
eb.setOptLevel(llvm.CodeGenOpt.Level.Aggressive).setUseMCJIT(False)
tm = eb.selectTarget()
print('target triple:', tm.getTargetTriple())
print('target cpu:', tm.getTargetCPU())
print('target feature string:', tm.getTargetFeatureString())
target = tm.getTarget()
print('target name:', target.getName())
print('target short description:', target.getShortDescription())
assert target.hasJIT()
assert target.hasTargetMachine()
ee = eb.create(tm)
triple = llvm.Triple.new('x86_64-unknown-linux')
assert triple.getArchName() == 'x86_64'
assert triple.getVendorName() == 'unknown'
assert triple.getOSName() == 'linux'
assert triple.isArch64Bit()
assert not triple.isArch32Bit()
triple_32variant = triple.get32BitArchVariant()
assert triple_32variant.isArch32Bit()
print(tm.getDataLayout())
pm = llvm.PassManager.new()
pm.add(llvm.DataLayout.new(str(tm.getDataLayout())))
pm.add(llvm.TargetLibraryInfo.new())
# write assembly
pm = llvm.PassManager.new()
pm.add(llvm.DataLayout.new(str(tm.getDataLayout())))
raw = extra.make_raw_ostream_for_printing()
formatted = llvm.formatted_raw_ostream.new(raw, False)
cgft = llvm.TargetMachine.CodeGenFileType.CGFT_AssemblyFile
failed = tm.addPassesToEmitFile(pm, formatted, cgft, False)
assert not failed
pm.run(m)
formatted.flush()
raw.flush()
asm = raw.str()
print(asm)
assert 'foo' in asm
def test_basic_jit_use():
context = llvm.getGlobalContext()
m = llvm.Module.new("modname", context)
print(m.getModuleIdentifier())
m.setModuleIdentifier('modname2')
print(m.getModuleIdentifier())
print('endianness', m.getEndianness())
assert m.getEndianness() == llvm.Module.Endianness.AnyEndianness
print('pointer-size', m.getPointerSize())
assert m.getPointerSize() == llvm.Module.PointerSize.AnyPointerSize
m.dump()
os = extra.make_raw_ostream_for_printing()
m.print_(os, None)
print(os.str())
int1ty = llvm.Type.getInt1Ty(context)
int1ty.dump()
assert int1ty.isIntegerTy(1)
fnty = llvm.FunctionType.get(int1ty, False)
fnty.dump()
types = [llvm.Type.getIntNTy(context, 8), llvm.Type.getIntNTy(context, 32)]
fnty = llvm.FunctionType.get(llvm.Type.getIntNTy(context, 8), types, False)
print(fnty)
const = m.getOrInsertFunction("foo", fnty)
fn = const._downcast(llvm.Function)
print(fn)
assert fn.hasName()
assert 'foo' == fn.getName()
fn.setName('bar')
assert 'bar' == fn.getName()
assert fn.getReturnType().isIntegerTy(8)
assert fnty is fn.getFunctionType()
assert fn.isVarArg() == False
assert fn.getIntrinsicID() == 0
assert not fn.isIntrinsic()
fn_uselist = fn.list_use()
assert isinstance(fn_uselist, list)
assert len(fn_uselist) == 0
builder = llvm.IRBuilder.new(context)
print(builder)
bb = llvm.BasicBlock.Create(context, "entry", fn, None)
assert bb.empty()
builder.SetInsertPoint(bb)
assert bb.getTerminator() is None
arg0, arg1 = fn.getArgumentList()
print(arg0, arg1)
extended = builder.CreateZExt(arg0, arg1.getType())
result = builder.CreateAdd(extended, arg1)
ret = builder.CreateTrunc(result, fn.getReturnType())
builder.CreateRet(ret)
print(arg0.list_use())
print(fn)
errio = StringIO()
print(m)
# verifier
action = llvm.VerifierFailureAction.ReturnStatusAction
corrupted = llvm.verifyFunction(fn, action)
assert not corrupted
corrupted = llvm.verifyModule(m, action, errio)
print(corrupted)
assert not corrupted, errio.getvalue()
# build pass manager
pmb = llvm.PassManagerBuilder.new()
pmb.OptLevel = 3
assert pmb.OptLevel == 3
pmb.LibraryInfo = llvm.TargetLibraryInfo.new()
pmb.Inliner = llvm.createFunctionInliningPass()
fpm = llvm.FunctionPassManager.new(m)
pm = llvm.PassManager.new()
pmb.populateFunctionPassManager(fpm)
pmb.populateModulePassManager(pm)
fpm.doInitialization()
fpm.run(fn)
fpm.doFinalization()
pm.run(m)
print(m)
# build engine
ee = llvm.ExecutionEngine.createJIT(m, errio)
print(ee, errio.getvalue())
print(ee.getDataLayout().getStringRepresentation())
datalayout_str = 'e-p:64:64:64-S128-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f16:16:16-f32:32:32-f64:64:64-f128:128:128-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64'
assert datalayout_str == str(llvm.DataLayout.new(datalayout_str))
assert datalayout_str == str(llvm.DataLayout.new(str(llvm.DataLayout.new(datalayout_str))))
fn2 = ee.FindFunctionNamed(fn.getName())
assert fn2 is fn
assert ee.getPointerToFunction(fn)
assert ee.getPointerToNamedFunction('printf')
gv0 = llvm.GenericValue.CreateInt(arg0.getType(), 12, False)
gv1 = llvm.GenericValue.CreateInt(arg1.getType(), -32, True)
assert gv0.valueIntWidth() == arg0.getType().getIntegerBitWidth()
assert gv1.valueIntWidth() == arg1.getType().getIntegerBitWidth()
assert gv0.toUnsignedInt() == 12
assert gv1.toSignedInt() == -32
gv1 = llvm.GenericValue.CreateInt(arg1.getType(), 32, False)
gvR = ee.runFunction(fn, (gv0, gv1))
assert 44 == gvR.toUnsignedInt()
# write bitcode
bc_buffer = BytesIO()
llvm.WriteBitcodeToFile(m, bc_buffer)
bc = bc_buffer.getvalue()
bc_buffer.close()
# read bitcode
errbuf = BytesIO()
m2 = llvm.ParseBitCodeFile(bc, context, errbuf)
if not m2:
raise Exception(errbuf.getvalue())
else:
m2.setModuleIdentifier(m.getModuleIdentifier())
assert str(m2) == str(m)
# parse llvm ir
m3 = llvm.ParseAssemblyString(str(m), None, llvm.SMDiagnostic.new(), context)
m3.setModuleIdentifier(m.getModuleIdentifier())
assert str(m3) == str(m)
# test clone
m4 = llvm.CloneModule(m)
assert m4 is not m
assert str(m4) == str(m)
def __str__(self):
from llvmpy import extra
os = extra.make_raw_ostream_for_printing()
self.print_(os, None)
return os.str()
def test_engine_builder():
llvm.InitializeNativeTarget()
context = llvm.getGlobalContext()
m = llvm.Module.new("modname", context)
int32ty = llvm.Type.getIntNTy(context, 32)
fnty = llvm.FunctionType.get(int32ty, [int32ty], False)
fn = m.getOrInsertFunction("foo", fnty)._downcast(llvm.Function)
bb = llvm.BasicBlock.Create(context, "entry", fn, None)
builder = llvm.IRBuilder.new(context)
builder.SetInsertPoint(bb)
builder.CreateRet(fn.getArgumentList()[0])
print(fn)
eb = llvm.EngineBuilder.new(m)
eb2 = eb.setEngineKind(llvm.EngineKind.Kind.JIT)
assert eb is eb2
eb.setOptLevel(llvm.CodeGenOpt.Level.Aggressive).setUseMCJIT(False)
tm = eb.selectTarget()
print('target triple:', tm.getTargetTriple())
print('target cpu:', tm.getTargetCPU())
print('target feature string:', tm.getTargetFeatureString())
target = tm.getTarget()
print('target name:', target.getName())
print('target short description:', target.getShortDescription())
assert target.hasJIT()
assert target.hasTargetMachine()
ee = eb.create(tm)
triple = llvm.Triple.new('x86_64-unknown-linux')
assert triple.getArchName() == 'x86_64'
assert triple.getVendorName() == 'unknown'
assert triple.getOSName() == 'linux'
assert triple.isArch64Bit()
assert not triple.isArch32Bit()
triple_32variant = triple.get32BitArchVariant()
assert triple_32variant.isArch32Bit()
print(tm.getDataLayout())
pm = llvm.PassManager.new()
pm.add(llvm.DataLayout.new(str(tm.getDataLayout())))
pm.add(llvm.TargetLibraryInfo.new())
# write assembly
pm = llvm.PassManager.new()
pm.add(llvm.DataLayout.new(str(tm.getDataLayout())))
raw = extra.make_raw_ostream_for_printing()
formatted = llvm.formatted_raw_ostream.new(raw, False)
cgft = llvm.TargetMachine.CodeGenFileType.CGFT_AssemblyFile
failed = tm.addPassesToEmitFile(pm, formatted, cgft, False)
assert not failed
pm.run(m)
formatted.flush()
raw.flush()
asm = raw.str()
print(asm)
assert 'foo' in asm
def test_basic_jit_use():
context = llvm.getGlobalContext()
m = llvm.Module.new("modname", context)
print(m.getModuleIdentifier())
m.setModuleIdentifier('modname2')
print(m.getModuleIdentifier())
print('endianness', m.getEndianness())
assert m.getEndianness() == llvm.Module.Endianness.AnyEndianness
print('pointer-size', m.getPointerSize())
assert m.getPointerSize() == llvm.Module.PointerSize.AnyPointerSize
m.dump()
os = extra.make_raw_ostream_for_printing()
m.print_(os, None)
print(os.str())
int1ty = llvm.Type.getInt1Ty(context)
int1ty.dump()
assert int1ty.isIntegerTy(1)
fnty = llvm.FunctionType.get(int1ty, False)
fnty.dump()
types = [llvm.Type.getIntNTy(context, 8), llvm.Type.getIntNTy(context, 32)]
fnty = llvm.FunctionType.get(llvm.Type.getIntNTy(context, 8), types, False)
print(fnty)
const = m.getOrInsertFunction("foo", fnty)
fn = const._downcast(llvm.Function)
print(fn)
assert fn.hasName()
assert 'foo' == fn.getName()
fn.setName('bar')
assert 'bar' == fn.getName()
assert fn.getReturnType().isIntegerTy(8)
assert fnty is fn.getFunctionType()
assert fn.isVarArg() == False
assert fn.getIntrinsicID() == 0
assert not fn.isIntrinsic()
fn_uselist = fn.list_use()
assert isinstance(fn_uselist, list)
assert len(fn_uselist) == 0
builder = llvm.IRBuilder.new(context)
print(builder)
bb = llvm.BasicBlock.Create(context, "entry", fn, None)
assert bb.empty()
builder.SetInsertPoint(bb)
assert bb.getTerminator() is None
arg0, arg1 = fn.getArgumentList()
print(arg0, arg1)
extended = builder.CreateZExt(arg0, arg1.getType())
result = builder.CreateAdd(extended, arg1)
ret = builder.CreateTrunc(result, fn.getReturnType())
builder.CreateRet(ret)
print(arg0.list_use())
print(fn)
errio = StringIO()
print(m)
# verifier
action = llvm.VerifierFailureAction.ReturnStatusAction
corrupted = llvm.verifyFunction(fn, action)
assert not corrupted
corrupted = llvm.verifyModule(m, action, errio)
print(corrupted)
assert not corrupted, errio.getvalue()
# build pass manager
pmb = llvm.PassManagerBuilder.new()
pmb.OptLevel = 3
assert pmb.OptLevel == 3
pmb.LibraryInfo = llvm.TargetLibraryInfo.new()
pmb.Inliner = llvm.createFunctionInliningPass()
fpm = llvm.FunctionPassManager.new(m)
pm = llvm.PassManager.new()
pmb.populateFunctionPassManager(fpm)
pmb.populateModulePassManager(pm)
fpm.doInitialization()
fpm.run(fn)
fpm.doFinalization()
pm.run(m)
print(m)
# build engine
ee = llvm.ExecutionEngine.createJIT(m, errio)
print(ee, errio.getvalue())
print(ee.getDataLayout().getStringRepresentation())
datalayout_str = 'e-p:64:64:64-S128-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f16:16:16-f32:32:32-f64:64:64-f128:128:128-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64'
assert datalayout_str == str(llvm.DataLayout.new(datalayout_str))
assert datalayout_str == str(
llvm.DataLayout.new(str(llvm.DataLayout.new(datalayout_str))))
fn2 = ee.FindFunctionNamed(fn.getName())
assert fn2 is fn
assert ee.getPointerToFunction(fn)
assert ee.getPointerToNamedFunction('printf')
gv0 = llvm.GenericValue.CreateInt(arg0.getType(), 12, False)
gv1 = llvm.GenericValue.CreateInt(arg1.getType(), -32, True)
assert gv0.valueIntWidth() == arg0.getType().getIntegerBitWidth()
assert gv1.valueIntWidth() == arg1.getType().getIntegerBitWidth()
assert gv0.toUnsignedInt() == 12
assert gv1.toSignedInt() == -32
gv1 = llvm.GenericValue.CreateInt(arg1.getType(), 32, False)
gvR = ee.runFunction(fn, (gv0, gv1))
assert 44 == gvR.toUnsignedInt()
# write bitcode
bc_buffer = BytesIO()
llvm.WriteBitcodeToFile(m, bc_buffer)
bc = bc_buffer.getvalue()
bc_buffer.close()
# read bitcode
errbuf = BytesIO()
m2 = llvm.ParseBitCodeFile(bc, context, errbuf)
if not m2:
raise Exception(errbuf.getvalue())
else:
m2.setModuleIdentifier(m.getModuleIdentifier())
assert str(m2) == str(m)
# parse llvm ir
m3 = llvm.ParseAssemblyString(str(m), None, llvm.SMDiagnostic.new(),
context)
m3.setModuleIdentifier(m.getModuleIdentifier())
assert str(m3) == str(m)
# test clone
m4 = llvm.CloneModule(m)
assert m4 is not m
assert str(m4) == str(m)
def __str__(self):
os = extra.make_raw_ostream_for_printing()
self.tm.inst_printer.printInst(self.mcinst, os, "")
return str(os.str())
def __str__(self):
os = extra.make_raw_ostream_for_printing()
self.tm.inst_printer.printInst(self.mcinst, os, "")
return str(os.str())
