AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-343921 - added Max stack size

Browse Source 
Change-Id: I5c1a088e05215ca951afc9d92f8d298c5e3a65f1


[ROCm/clr commit: 27a08a132f]
This commit is contained in:
Sarbojit Sarkar authored and Sarbojit Sarkar committed 2022-08-02 07:13:18 -04:00
1 parent 1d2cb0d3e8
commit a0981a092b
3 files changed
+7 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/clr/rocclr/device/rocm/rocvirtual.cpp
+4 -4
View File
@@ -2972,10 +2972,10 @@ bool VirtualGPU::submitKernelInternal(const amd::NDRangeContainer& sizes,
dispatchPacket.private_segment_size = devKernel->workGroupInfo()->privateMemSize_;
if ((devKernel->workGroupInfo()->usedStackSize_ & 0x1) == 0x1) {
dispatchPacket.private_segment_size += dev().StackSize();
uint32_t maxMemPerThread = device().info().localMemSizePerCU_ / device().info().maxThreadsPerCU_;
if (dispatchPacket.private_segment_size > maxMemPerThread) {
dispatchPacket.private_segment_size = maxMemPerThread;
dispatchPacket.private_segment_size =
std::max<uint64_t>(dev().StackSize(), dispatchPacket.private_segment_size);
if (dispatchPacket.private_segment_size > 16 * Ki) {
dispatchPacket.private_segment_size = 16 * Ki;
}
}