triton
triton copied to clipboard
[AMD] Fix kwidth printing and parsing
This PR fixes:
- parsing and printing of kWidth attribute for MFMA and WMMA layouts
- comment in 06-fused-attention.p tutorial
FYI @oplavsic
Also, @binarman, please make sure all tests pass before marking pull requests as ready for review. As addressing bots happy needs changing the impl. (You can ping me if you need to trigger CI runs.)
@antiagainst @jlebar I've updated this PR, can you take a look?
(converting to draft as we said internally. will mark it open after approval from me)