search

triton-lang / triton

Development repository for the Triton language and compiler
https://triton-lang.org/
MIT License
11.84k stars 1.39k forks source link

[AOT] is there a way we can apply AOT in AMD environment? #3536

Open xinji1 opened 3 months ago

xinji1 commented 3 months ago

For the test in /triton/python/test/unit/tools/test_aot.py, there's an error like: image

Traceback (most recent call last):                                                                                           
  File "/python3.10/site-packages/triton/tools/compile.py", line 119, in <module>
    hex_ = str(binascii.hexlify(ccinfo.asm["cubin"]))[2:-1]                                                                                                                                                                                               
KeyError: 'cubin'

So is there a way we can apply AOT in AMD(here is AMD Instinct MI250X/MI250) environment?

xinji1 commented 3 months ago

Updated: just found a possible solution https://github.com/ROCm/aotriton