mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-13 17:08:11 -05:00
32 lines
2.3 KiB
Plaintext
32 lines
2.3 KiB
Plaintext
An integrated environment for AMD GPU assembly and emulation
|
|
|
|
Test with `PYTHONPATH="." pytest -n12 extra/assembly/amd/`
|
|
`AMD_LLVM=1 PYTHONPATH="." pytest -n12 extra/assembly/amd/`
|
|
|
|
* pdf.py -- extract assembly format + instruction psuedocode from AMD PDF
|
|
* dsl.py -- helpers for the autogen instruction classes in `__init__.py`. should be standalone with init
|
|
* pcode.py -- psuedocode execution environment. psuedocode should be transformed as little as possible.
|
|
* asm.py -- an asm/disasm function to transform to and from AMD assembly syntax
|
|
* emu.py -- an emulator for RDNA that runs in tinygrad with `AMD=1 MOCKGPU=1 PYTHON_REMU=1`
|
|
|
|
The code should be as readable and deduplicated as possible. asm and emu shouldn't be required for dsl.
|
|
|
|
test_emu.py has a good set of instruction tests for the emulation, with USE_HW=1 it will compare to real hardware.
|
|
Whenever an instruction is fixed, regression tests should be added here and confirmed with real hardware.
|
|
|
|
test_llvm.py tests asm/disasm on the LLVM tests, confirming it behaves the same as LLVM.
|
|
|
|
tinygrad's dtype tests should pass with and without LLVM. they run in about 12 seconds.
|
|
|
|
`PYTHONPATH="." AMD=1 PYTHON_REMU=1 MOCKGPU=1 AMD_LLVM=0 pytest -n=12 test/test_dtype_alu.py test/test_dtype.py`
|
|
`PYTHONPATH="." AMD=1 PYTHON_REMU=1 MOCKGPU=1 AMD_LLVM=1 pytest -n=12 test/test_dtype_alu.py test/test_dtype.py`
|
|
|
|
The ops tests also pass, but they are very slow, so you should run them one at a time.
|
|
|
|
`SKIP_SLOW_TEST=1 PYTHONPATH="." AMD=1 PYTHON_REMU=1 MOCKGPU=1 AMD_LLVM=0 pytest -n=12 test/test_ops.py`
|
|
`SKIP_SLOW_TEST=1 PYTHONPATH="." AMD=1 PYTHON_REMU=1 MOCKGPU=1 AMD_LLVM=1 pytest -n=12 test/test_ops.py`
|
|
|
|
When something is caught by main tinygrad tests, a local regression test should be added to `extra/assembly/amd/test`. While working with tinygrad, you can dump the assembly with `DEBUG=7`. These tests all pass on real hardware, so if a test is failing with `AMD=1 PYTHON_REMU=1 MOCKGPU=1` it's likely because an instruction is emulated incorrectly. You can test without `MOCKGPU=1` to test on real hardware, if it works on real hardware there's a bug in the emulator.
|
|
|
|
Currently, only RDNA3 is well supported, but when finished, this will support RDNA3+RDNA4+CDNA in ~2000 lines. Count lines with `cloc --by-file extra/assembly/amd/*.py`
|