Vehicle Researcher
6adb63b915
date: 2026-06-04T09:49:56 master commit: c0ab3550eca2e9daf197c46b7e4b24aa9637cf2e
39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
import unittest, subprocess, platform
|
|
from tinygrad.runtime.support.compiler_cpu import ClangJITCompiler
|
|
from tinygrad.runtime.support.elf import elf_loader
|
|
|
|
class TestElfLoader(unittest.TestCase):
|
|
def test_load_clang_jit_strtab(self):
|
|
src = '''
|
|
int something; // will be a load from a relocation (needed for .rela.text to exist)
|
|
int test(int x) {
|
|
return something + x;
|
|
}
|
|
'''
|
|
args = ('-x', 'c', '-c', '-target', f'{platform.machine()}-none-unknown-elf', '-march=native', '-fPIC', '-O2', '-ffreestanding', '-nostdlib')
|
|
obj = subprocess.check_output(('clang',) + args + ('-', '-o', '-'), input=src.encode('utf-8'))
|
|
_, sections, _ = elf_loader(obj)
|
|
section_names = [sh.name for sh in sections]
|
|
assert '.text' in section_names and '.rela.text' in section_names, str(section_names)
|
|
def test_clang_jit_compiler_external_raise(self):
|
|
src = '''
|
|
int evil_external_function(int);
|
|
int test(int x) {
|
|
return evil_external_function(x+2)*2;
|
|
}
|
|
'''
|
|
with self.assertRaisesRegex(RuntimeError, 'evil_external_function'):
|
|
ClangJITCompiler([{'AMD64':'x86_64', 'aarch64':'arm64'}.get(m:=platform.machine(), m), "native"]).compile(src)
|
|
def test_link(self):
|
|
src = '''
|
|
float powf(float, float); // from libm
|
|
float test(float x, float y) { return powf(x, y); }
|
|
'''
|
|
args = ('-x', 'c', '-c', '-target', f'{platform.machine()}-none-unknown-elf', '-march=native', '-fPIC', '-O2', '-ffreestanding', '-nostdlib')
|
|
obj = subprocess.check_output(('clang',) + args + ('-', '-o', '-'), input=src.encode())
|
|
with self.assertRaisesRegex(RuntimeError, 'powf'): elf_loader(obj)
|
|
elf_loader(obj, link_libs=['m'])
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|