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

SWDEV-459758 - Pass workgroup size explicitly (#185)

Browse Source 
It's easier for compiler to move explicit kernel arguments into user SGPRs
This commit is contained in:
Andryeyev, German
2025-04-15 15:22:15 -04:00
committed by GitHub
parent 5d638d831c
commit 3fd7650fe3
3 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rocclr/device/pal/palblit.cpp
+4
View File
@@ -2222,6 +2222,8 @@ bool KernelBlitManager::fillBuffer(device::Memory& memory, const void* pattern,
setArgument(kernels_[kFillType], 4, sizeof(end_ptr), &end_ptr);
uint32_t next_chunk = globalWorkSize * kpattern_size;
setArgument(kernels_[kFillType], 5, sizeof(uint32_t), &next_chunk);
uint32_t lws = localWorkSize;
setArgument(kernels_[kFillType], 6, sizeof(lws), &lws);
// Create ND range object for the kernel's execution
amd::NDRangeContainer ndrange(1, globalWorkOffset, &globalWorkSize, &localWorkSize);
@@ -2285,6 +2287,8 @@ bool KernelBlitManager::copyBuffer(device::Memory& srcMemory, device::Memory& ds
uint32_t next_chunk = globalWorkSize;
setArgument(kernels_[kBlitType], 6, sizeof(next_chunk), &next_chunk);
uint32_t lws = localWorkSize;
setArgument(kernels_[kBlitType], 7, sizeof(lws), &lws);
// Create ND range object for the kernel's execution
amd::NDRangeContainer ndrange(1, nullptr, &globalWorkSize, &localWorkSize);