mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 15:08:02 -05:00
0aabc1e938
* nak works * TestOps::test_add works * testop has no crashes * fix bool casts * fix typo * add disassemble * RANGE and locals/regs * simplify NAKCompiler * disass cleanup * cleanup nir codegen * almost all tests passing * cleanup notes in extra/ * old notes * only import nak if NIR=1 * fix new SPECIAL syntax * fix local/shared memory * more tests passing * add DEFINE_VAR support * llvmpipe kinda works * diskcache * some mypy stuff * lvp passing test_ops.py * fix imports * actually fix imports * remove 'stdout' * fix llvm import * fix mypy issues * nicer errors * simpler test_dtype skips * test lvp in CI * fix github action syntax * fix more actions typos * switch to mesa 25.1.0 * diskcache_put * better generation for lvp nir_options * b64encode shader blobs * Revert diskcache changes This reverts commits930fa3de8aand8428c694b3. * general cleanup * better error messages * fix llvm import * fix windows tests * link with libm and libgcc_s * fix some errors * dont check for 'float4' * NIR uses pointer arithmetic * use tinymesa * bump tinymesa * bump tinymesa again * update lvp nir_options * print nir shader with DEBUG * simplify LVPCompiler * more tests * "gated" STORE * NAK is cacheable * more tests * all tests pass locally for NAK * test autogen in CI * autogen deps * more deps * fix uop_gc * fix macos * mypy * save 2 lines * save two more lines * save 1 line * save 4 lines * save more lines * Revert "save more lines" This reverts commitdd3a720c5a. * save more lines * fix LVP on windows * refactor * reorganize some code * refactor lib_gpu * move LVP check * out of order loads * remove support.mesa * bump tinymesa version * simplify LVP jit * macos * macos ci * shell: bash * testing * more testing * compute brew prefix * stupid typo * actually fix * lib * stdout on macos * inline gallivm_compile_module * Revert "inline gallivm_compile_module" This reverts commitb65983b151. * elf macos * semicolon * inherit from CPULLVMCompiler * ruff * disas test * fix libm linking * default is fine actually * arm works * add elf loader link test * fix NAK beam * pylint is too smart by half --------- Co-authored-by: George Hotz <72895+geohot@users.noreply.github.com> Co-authored-by: nimlgen <138685161+nimlgen@users.noreply.github.com>
34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
import unittest, io
|
|
from contextlib import redirect_stdout
|
|
from tinygrad import Tensor, dtypes, Device
|
|
from tinygrad.helpers import OSX, CPU_LLVM, CPU_LVP
|
|
from tinygrad.engine.realize import lower_schedule
|
|
from tinygrad.device import is_dtype_supported
|
|
from tinygrad.engine.realize import get_program
|
|
|
|
class TestCompileFailures(unittest.TestCase):
|
|
def compile(self, out:Tensor):
|
|
for _ in lower_schedule(out.schedule()): pass
|
|
|
|
@unittest.skipUnless(is_dtype_supported(dtypes.uchar, Device.DEFAULT), f"no uint8 on {Device.DEFAULT}")
|
|
def test_interpolate_atari(self):
|
|
self.compile(Tensor.empty(210, 160, dtype='uint8').interpolate((64, 64)))
|
|
|
|
def test_add_max_uchar(self):
|
|
self.compile((Tensor.empty(1024, dtype='uint8') + Tensor.empty(1024, dtype='uint8')).max())
|
|
|
|
class TestDisassembly(unittest.TestCase):
|
|
# TODO: fails on llvm. llvm.LLVMGetHostCPUName() returns "generic"
|
|
@unittest.skipUnless(Device.DEFAULT in ("CPU",) and not (CPU_LLVM or CPU_LVP) and OSX, "m series cpus support fp16 arithmetic")
|
|
def test_float16_alu(self):
|
|
c = Tensor([1], dtype=dtypes.float16) + Tensor([1], dtype=dtypes.float16)
|
|
s = c.schedule()[-1]
|
|
p = get_program(s.ast, Device[Device.DEFAULT].renderer)
|
|
lib = Device[Device.DEFAULT].compiler.compile(p.src)
|
|
out = io.StringIO()
|
|
with redirect_stdout(out): Device[Device.DEFAULT].compiler.disassemble(lib)
|
|
assert "fcvt" not in out.getvalue()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|