enable register usage and spill for AMD backend

ROCm / triton

Development repository for the Triton language and compiler

MIT License

80 stars 22 forks source link

enable register usage and spill for AMD backend #535

Closed xiaohuguo2023 closed 3 months ago

xiaohuguo2023 commented 3 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")