Skip to content

Commit d81b140

Browse files
dmitryryintelsys_zuul
authored andcommitted
Support private base in ZEBinary
Change-Id: Ic678fa260ec94e5076139fead3f1f465d90d3014
1 parent 4fda952 commit d81b140

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

IGC/AdaptorOCL/cmc.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff 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

221223
void 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;

0 commit comments

Comments
 (0)