enable register usage and spill for AMD backend

ROCm / triton

Development repository for the Triton language and compiler

MIT License

89 stars 27 forks source link

enable register usage and spill for AMD backend #536

Closed xiaohuguo2023 closed 6 months ago

xiaohuguo2023 commented 6 months ago

This is trying to enable our AMD backend to print out kernel register usage information like below

grids = total_programs_streamk + total_blocking_tiles
kk = theKernel[(grids,)]()
if DEBUG:
     print(f"{kk.n_regs} registers used, {kk.n_spills} spills")