mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix llvm 20 bug

Browse source
This commit is contained in:
George Hotz 2025-05-04 17:26:44 -07:00
commit edc1e053fa
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tinygrad/runtime/ops_llvm.py
View file

@ -1,6 +1,6 @@
import functools, ctypes, platform
from tinygrad.device import Compiled, Compiler, MallocAllocator, CPUProgram
from tinygrad.helpers import OSX, getenv, capstone_flatdump, DEBUG
from tinygrad.helpers import OSX, getenv, capstone_flatdump, DEBUG, to_char_p_p
from tinygrad.renderer.llvmir import LLVMRenderer
import tinygrad.runtime.autogen.llvm as llvm
from tinygrad.runtime.support.elf import jit_loader
@ -17,6 +17,9 @@ class LLVMCompiler(Compiler):
def __init__(self, processor:str, feats:str):
for component in ['Target', 'TargetInfo', 'TargetMC', 'AsmParser', 'AsmPrinter']: getattr(llvm, f'LLVMInitialize{self.target_arch}{component}')()
# NOTE: fix `TestOptGemm.test_gemm_unrolled_permute_l_42` bug in LLVM 20
llvm.LLVMParseCommandLineOptions(2, to_char_p_p([b"llvm", b"-enable-constraint-elimination=0"]), None)
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 >= 2: print(f"LLVM init for {processor!r} with {feats!r}")