Skip to content

Commit 772ae05

Browse files
committed
Use urKernelGetSuggestedLocalWorkSize
1 parent ee525e1 commit 772ae05

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

source/loader/layers/sanitizer/asan_interceptor.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -767,10 +767,9 @@ ur_result_t SanitizerInterceptor::prepareLaunch(
767767

768768
if (LaunchInfo.LocalWorkSize.empty()) {
769769
LaunchInfo.LocalWorkSize.reserve(3);
770-
// FIXME: This is W/A until urKernelSuggestGroupSize is added
771-
LaunchInfo.LocalWorkSize[0] = 1;
772-
LaunchInfo.LocalWorkSize[1] = 1;
773-
LaunchInfo.LocalWorkSize[2] = 1;
770+
context.urDdiTable.Kernel.pfnGetSuggestedLocalWorkSize(
771+
Kernel, Queue, LaunchInfo.WorkDim, LaunchInfo.GlobalWorkOffset,
772+
LaunchInfo.GlobalWorkSize, LaunchInfo.LocalWorkSize.data());
774773
}
775774

776775
const size_t *LocalWorkSize = LaunchInfo.LocalWorkSize.data();

0 commit comments

Comments
 (0)