[llvm] [AMDGPU] Update code object metadata for kernarg preload (PR #134666)

Shilei Tian via llvm-commits llvm-commits at lists.llvm.org
Mon Jun 23 09:59:48 PDT 2025


================
@@ -95,11 +98,79 @@ inline raw_ostream &operator<<(raw_ostream &OS, const ArgDescriptor &Arg) {
   return OS;
 }
 
-struct KernArgPreloadDescriptor : public ArgDescriptor {
-  KernArgPreloadDescriptor() {}
-  SmallVector<MCRegister> Regs;
+namespace KernArgPreload {
+
+enum HiddenArg {
+  HIDDEN_BLOCK_COUNT_X = 0,
+  HIDDEN_BLOCK_COUNT_Y = 1,
+  HIDDEN_BLOCK_COUNT_Z = 2,
+  HIDDEN_GROUP_SIZE_X = 3,
+  HIDDEN_GROUP_SIZE_Y = 4,
+  HIDDEN_GROUP_SIZE_Z = 5,
+  HIDDEN_REMAINDER_X = 6,
+  HIDDEN_REMAINDER_Y = 7,
+  HIDDEN_REMAINDER_Z = 8,
+  END_HIDDEN_ARGS = 9
----------------
shiltian wrote:

nit: `HIDDEN_REMAINDER_Z + 1` might be better here for future maintenance?

https://github.com/llvm/llvm-project/pull/134666


More information about the llvm-commits mailing list