efee1712aa
* fix.. speed_limit error... * draw tpms settings. * fix.. traffic light stopping only.. * fix.. waze cam * fix.. waze... * add setting (Enable comma connect ) * auto detect LFA2 * fix.. cruisespeed1 * vff2 driving model. * fix.. * agnos 12.3 * fix.. * ff * ff * test * ff * fix.. drawTurnInfo.. * Update drive_helpers.py * fix.. support eng voice eng sounds fix settings... english fix.. mph.. fix.. roadlimit speed bug.. * new vff model.. 250608 * fix soundd.. * fix safe exit speed.. * fix.. sounds. * fix.. radar timeStep.. * KerryGold model * Update drive_helpers.py * fix.. model. * fix.. * fix.. * Revert "fix.." This reverts commit b09ec459afb855c533d47fd7e8a1a6b1a09466e7. * Revert "fix.." This reverts commit 290bec6b83a4554ca232d531a911edccf94a2156. * fix esim * add more acc table. 10kph * kg update.. * fix cruisebutton mode3 * test atc..cond. * fix.. canfd * fix.. angle control limit
76 lines
4.3 KiB
Python
76 lines
4.3 KiB
Python
import functools, ctypes, platform
|
|
from tinygrad.device import Compiled, Compiler, MallocAllocator, CPUProgram
|
|
from tinygrad.helpers import OSX, getenv, capstone_flatdump, DEBUG
|
|
from tinygrad.renderer.llvmir import LLVMRenderer
|
|
import tinygrad.runtime.autogen.llvm as llvm
|
|
from tinygrad.runtime.support.elf import jit_loader
|
|
|
|
def cerr(): return ctypes.pointer(ctypes.pointer(ctypes.c_char()))
|
|
|
|
def expect(x, err, ret=None):
|
|
if x: raise RuntimeError(llvm.string_cast(err.contents) if not isinstance(err, str) else err)
|
|
return ret
|
|
|
|
class LLVMCompiler(Compiler):
|
|
jit = True
|
|
target_arch = {'arm64': 'AArch64', 'aarch64': 'AArch64', 'x86_64': 'X86', 'AMD64': 'X86'}[platform.machine()]
|
|
def __init__(self, processor:str, feats:str):
|
|
for component in ['Target', 'TargetInfo', 'TargetMC', 'AsmParser', 'AsmPrinter']: getattr(llvm, f'LLVMInitialize{self.target_arch}{component}')()
|
|
|
|
triple = {'AArch64': b'aarch64-none-unknown-elf', 'X86': b'x86_64-none-unknown-elf', 'AMDGPU': b'amdgcn-amd-amdhsa'}[self.target_arch]
|
|
target = expect(llvm.LLVMGetTargetFromTriple(triple, ctypes.pointer(tgt:=llvm.LLVMTargetRef()), err:=cerr()), err, tgt)
|
|
if DEBUG >= 3: print(f"LLVM init for {processor!r} with {feats!r}")
|
|
self.target_machine = llvm.LLVMCreateTargetMachine(target, triple, processor.encode(), feats.encode(),
|
|
llvm.LLVMCodeGenLevelDefault, llvm.LLVMRelocPIC, llvm.LLVMCodeModelDefault)
|
|
|
|
self.pbo = llvm.LLVMCreatePassBuilderOptions()
|
|
if (opt:=bool(getenv("LLVMOPT", "1"))):
|
|
self.passes = b'default<O2>'
|
|
llvm.LLVMPassBuilderOptionsSetLoopUnrolling(self.pbo, True)
|
|
llvm.LLVMPassBuilderOptionsSetLoopVectorization(self.pbo, True)
|
|
llvm.LLVMPassBuilderOptionsSetSLPVectorization(self.pbo, True)
|
|
llvm.LLVMPassBuilderOptionsSetVerifyEach(self.pbo, True)
|
|
else:
|
|
self.passes = b'default<O0>'
|
|
|
|
self.diag_msgs: list[str] = []
|
|
@ctypes.CFUNCTYPE(None, llvm.LLVMDiagnosticInfoRef, ctypes.c_void_p)
|
|
def handle_diag(diag_ref, _arg):
|
|
severity = llvm.LLVMGetDiagInfoSeverity(diag_ref)
|
|
msg = ctypes.string_at(llvm.LLVMGetDiagInfoDescription(diag_ref)).decode()
|
|
if severity == llvm.LLVMDSError:
|
|
self.diag_msgs.append(msg)
|
|
self.handle_diag = handle_diag
|
|
llvm.LLVMContextSetDiagnosticHandler(llvm.LLVMGetGlobalContext(), handle_diag, None)
|
|
super().__init__(f"compile_llvm_{self.target_arch}{'_jit' if self.jit else ''}{'_opt' if opt else ''}")
|
|
|
|
def __del__(self): llvm.LLVMDisposePassBuilderOptions(self.pbo)
|
|
|
|
def compile(self, src:str) -> bytes:
|
|
self.diag_msgs.clear()
|
|
src_buf = llvm.LLVMCreateMemoryBufferWithMemoryRangeCopy(ctypes.create_string_buffer(src_bytes:=src.encode()), len(src_bytes), b'src')
|
|
mod = expect(llvm.LLVMParseIRInContext(llvm.LLVMGetGlobalContext(), src_buf, ctypes.pointer(m:=llvm.LLVMModuleRef()), err:=cerr()), err, m)
|
|
expect(llvm.LLVMVerifyModule(mod, llvm.LLVMReturnStatusAction, err:=cerr()), err)
|
|
expect(llvm.LLVMRunPasses(mod, self.passes, self.target_machine, self.pbo), 'failed to run passes')
|
|
if DEBUG >= 7: print(ctypes.string_at(llvm.LLVMPrintModuleToString(mod)).decode())
|
|
obj_buf = expect(llvm.LLVMTargetMachineEmitToMemoryBuffer(self.target_machine, mod, llvm.LLVMObjectFile, err:=cerr(),
|
|
ctypes.pointer(buf:=llvm.LLVMMemoryBufferRef())), err, buf)
|
|
llvm.LLVMDisposeModule(mod)
|
|
obj = ctypes.string_at(llvm.LLVMGetBufferStart(obj_buf), llvm.LLVMGetBufferSize(obj_buf))
|
|
llvm.LLVMDisposeMemoryBuffer(obj_buf)
|
|
if self.diag_msgs: raise RuntimeError("llvm diagnostic: " + "\n".join(self.diag_msgs))
|
|
return jit_loader(obj) if self.jit else obj
|
|
|
|
def disassemble(self, lib:bytes): capstone_flatdump(lib)
|
|
|
|
class HostLLVMCompiler(LLVMCompiler):
|
|
def __init__(self):
|
|
# +reserve-x18 here does the same thing as -ffixed-x18 in ops_cpu.py, see comments there for why it's needed on arm osx
|
|
cpu, feats = ctypes.string_at(llvm.LLVMGetHostCPUName()), (b'+reserve-x18,' if OSX else b'') + ctypes.string_at(llvm.LLVMGetHostCPUFeatures())
|
|
super().__init__(cpu.decode(), feats.decode())
|
|
|
|
class LLVMDevice(Compiled):
|
|
def __init__(self, device:str):
|
|
from tinygrad.runtime.graph.cpu import LLVMGraph
|
|
super().__init__(device, MallocAllocator, LLVMRenderer(), HostLLVMCompiler(), CPUProgram, functools.partial(LLVMGraph, self))
|