mirror/rea
1
0
Fork 0
mirror of https://github.com/morluto/rea synced 2026-10-27 14:42:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(gpu): resolve extended CUBIN symbol sections

Browse Source 
Fixes #6600
This commit is contained in:
morluto
2026-10-23 18:07:42 +08:00
parent 0205f870db
commit 00a9e95de9
3 changed files with 143 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/gpu/GpuElfKernels.ts
+27 -1
View File
@@ -266,6 +266,8 @@ const cudaKernels = async (
rejectGpu("CUBIN symbol string table is invalid");
const names = await sectionBytes(source, namesSection!, index.budget);
const bytes = await sectionBytes(source, section, index.budget);
const symbolCount = bytes.length / width;
let extendedIndices: Buffer | null = null;
for (let offset = 0; offset < bytes.length; offset += width) {
index.budget.visit();
const symbol = readElfSymbol(
@@ -275,7 +277,31 @@ const cudaKernels = async (
);
// STO_CUDA_ENTRY distinguishes launchable kernels from ordinary device functions.
if (symbol.type !== 2 || (symbol.other & 0x10) === 0) continue;
const text = directory.sections[symbol.section_index];
let sectionIndex = symbol.section_index;
if (sectionIndex === 0xffff) {
if (extendedIndices === null) {
const matches = directory.sections.filter(
(row) => row.type === 18 && row.link === section.index,
);
if (matches.length !== 1)
rejectGpu(
"CUBIN extended symbol section-index table is unavailable",
);
const extended = matches[0]!;
if (
extended.entry_size !== 4n ||
extended.size !== BigInt(symbolCount) * 4n
)
rejectGpu(
"CUBIN extended symbol section-index table has an invalid extent",
);
extendedIndices = await sectionBytes(source, extended, index.budget);
}
sectionIndex = new BinaryReader(extendedIndices, directory.little).u32(
(offset / width) * 4,
);
}
const text = directory.sections[sectionIndex];
if (text === undefined || text.type !== 1 || (text.flags & 4n) === 0n)
rejectGpu("CUBIN kernel symbol has no executable section");
const kernel = blankKernel(