File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -215,7 +215,9 @@ void CMKernel::createPrivateBaseAnnotation(
215215 m_kernelInfo.m_pointerInput .push_back (ptrAnnotation);
216216
217217 if (IGC_IS_FLAG_ENABLED (EnableZEBinary))
218- IGC_ASSERT_MESSAGE (0 , " not yet supported for L0 binary" );
218+ zebin::ZEInfoBuilder::addPayloadArgumentImplicit (m_kernelInfo.m_zePayloadArgs ,
219+ zebin::PreDefinedAttrGetter::ArgType::private_base_stateless,
220+ payloadPosition, byteSize);
219221}
220222
221223void CMKernel::createBufferStatefulAnnotation (unsigned argNo, cmc_access_kind accessKind)
@@ -464,6 +466,8 @@ static void generatePatchTokens_v2(const cmc_kernel_info_v2 *info,
464466 if (info->StatelessPrivateMemSize ) {
465467 kernel.createPrivateBaseAnnotation (AI.index , AI.sizeInBytes ,
466468 AI.offset , AI.BTI , info->StatelessPrivateMemSize );
469+ kernel.m_kernelInfo .m_executionEnivronment .PerThreadPrivateOnStatelessSize =
470+ info->StatelessPrivateMemSize ;
467471 kernel.m_kernelInfo .m_argIndexMap [AI.index ] = AI.BTI ;
468472 }
469473 break ;
You can’t perform that action at this time.
0 commit comments