ROCM IFU: Add new CTALayout parameter to mfma layout

This commit is contained in:
Aleksandr Efimov
2023-09-25 17:26:29 +00:00
committed by Jason Furmanek
parent e5d7bb4fae
commit bae0e4527c
3 changed files with 51 additions and 5 deletions

View File

@@ -718,7 +718,8 @@ The data will be distributed between threads as follows:
ins
"unsigned":$nonKDim,
ArrayRefParameter<"unsigned">:$warpsPerCTA,
"bool":$isTransposed
"bool":$isTransposed,
"CTALayoutAttr":$CTALayout
);
let hasCustomAssemblyFormat = 1;