Lines Matching defs:launch

129   // Identify uses from values defined outside of the scope of the launch
149     // Only replace uses within the launch op.
175 /// Outline the `gpu.launch` operation body into a kernel function. Replace
187   // Identify uses from values defined outside of the scope of the launch
207   // to the launch op, propagate them to the generated kernel. This is safe
218   // Map the arguments corresponding to the launch parameters like blockIdx,
235   // Map arguments from gpu.launch region to the arguments of the gpu.func
241   // Clone the region of the gpu.launch operation into the gpu.func operation.
255   // Splice now the entry block of the gpu.launch operation at the end of the
279 /// Replace `gpu.launch` operations with an `gpu.launch_func` operation
281 /// `gpu.launch` with constant region arguments inlined.
286   // The launch op has an optional dynamic shared memory size. If it doesn't
302 /// Pass that moves ops which are likely an index computation into gpu.launch
310     if (op->walk([](gpu::LaunchOp launch) {
312             if (failed(sinkOperationsIntoLaunchOp(launch,