Get ImplicitLocalArgPos from DeviceKernelInfo instead of ProgramManager (#20621)

vinser52 · web-flow · commit 33bb26d3dbbd · 2025-11-12T13:56:32.000-08:00
The performance regression was introduced in the #20316. We should get the position of the implicit local arg from the deviceKernelInfo, not from the ProgramManager.
diff --git a/sycl/source/detail/scheduler/commands.cpp b/sycl/source/detail/scheduler/commands.cpp
@@ -2470,8 +2470,7 @@ static ur_result_t SetKernelParamsAndLaunch(
   }
 
   std::optional<int> ImplicitLocalArg =
-      ProgramManager::getInstance().kernelImplicitLocalArgPos(
-          DeviceKernelInfo.Name);
+      DeviceKernelInfo.getImplicitLocalArgPos();
   // Set the implicit local memory buffer to support
   // get_work_group_scratch_memory. This is for backend not supporting
   // CUDA-style local memory setting. Note that we may have -1 as a position,

Original file line number	Diff line number	Diff line change
`@@ -2470,8 +2470,7 @@ static ur_result_t SetKernelParamsAndLaunch(`
`2470`	`2470`	`}`
`2471`	`2471`
`2472`	`2472`	`std::optional<int> ImplicitLocalArg =`
`2473`		`- ProgramManager::getInstance().kernelImplicitLocalArgPos(`
`2474`		`- DeviceKernelInfo.Name);`
	`2473`	`+ DeviceKernelInfo.getImplicitLocalArgPos();`
`2475`	`2474`	`// Set the implicit local memory buffer to support`
`2476`	`2475`	`// get_work_group_scratch_memory. This is for backend not supporting`
`2477`	`2476`	`// CUDA-style local memory setting. Note that we may have -1 as a position,`