Add live in for PrivateSegmentSize in GISel path (#139968)

jakemdaly · web-flow · commit 91b2e5bc5d3b · 2025-05-21T18:25:25.000+02:00
diff --git a/llvm/lib/Target/AMDGPU/AMDGPUCallLowering.cpp b/llvm/lib/Target/AMDGPU/AMDGPUCallLowering.cpp
@@ -493,6 +493,12 @@ static void allocateHSAUserSGPRs(CCState &CCInfo,
     CCInfo.AllocateReg(FlatScratchInitReg);
   }
 
+  if (UserSGPRInfo.hasPrivateSegmentSize()) {
+    Register PrivateSegmentSizeReg = Info.addPrivateSegmentSize(TRI);
+    MF.addLiveIn(PrivateSegmentSizeReg, &AMDGPU::SGPR_32RegClass);
+    CCInfo.AllocateReg(PrivateSegmentSizeReg);
+  }
+
   // TODO: Add GridWorkGroupCount user SGPRs when used. For now with HSA we read
   // these from the dispatch pointer.
 }

Original file line number	Diff line number	Diff line change
`@@ -493,6 +493,12 @@ static void allocateHSAUserSGPRs(CCState &CCInfo,`
`493`	`493`	`CCInfo.AllocateReg(FlatScratchInitReg);`
`494`	`494`	`}`
`495`	`495`
	`496`	`+ if (UserSGPRInfo.hasPrivateSegmentSize()) {`
	`497`	`+ Register PrivateSegmentSizeReg = Info.addPrivateSegmentSize(TRI);`
	`498`	`+ MF.addLiveIn(PrivateSegmentSizeReg, &AMDGPU::SGPR_32RegClass);`
	`499`	`+ CCInfo.AllocateReg(PrivateSegmentSizeReg);`
	`500`	`+ }`
	`501`	`+`
`496`	`502`	`// TODO: Add GridWorkGroupCount user SGPRs when used. For now with HSA we read`
`497`	`503`	`// these from the dispatch pointer.`
`498`	`504`	`}`