graphwise 1.9.1 → 1.11.0

package/dist/kernel-CvnRsF7E.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernel-CvnRsF7E.js","names":[],"sources":["../src/gpu/kernels/adamic-adar/kernel.ts"],"sourcesContent":["/**\n * TypeGPU compute kernel for Adamic-Adar index (batch).\n *\n * For each pair (u, v): iterate smaller neighbourhood, detect common neighbours\n * by binary search in the other neighbourhood, sum 1 / log(deg(w) + 1) and\n * normalise by log(2)/count to match CPU normalisation.\n */\n\nimport tgpu, { d, type StorageFlag, type TgpuBuffer } from \"typegpu\";\nimport type { TypedBufferGroup } from \"../../csr\";\nimport type { GraphwiseGPURoot } from \"../../root\";\n\nconst LOG2 = 0.6931471805599453;\n\nconst AdamicLayout = tgpu.bindGroupLayout({\n\trowOffsets: { storage: d.arrayOf(d.u32) },\n\tcolIndices: { storage: d.arrayOf(d.u32) },\n\tpairsU: { storage: d.arrayOf(d.u32) },\n\tpairsV: { storage: d.arrayOf(d.u32) },\n\tintersections: { storage: d.arrayOf(d.u32), access: \"mutable\" },\n\tsizeUs: { storage: d.arrayOf(d.u32), access: \"mutable\" },\n\tsizeVs: { storage: d.arrayOf(d.u32), access: \"mutable\" },\n\tresults: { storage: d.arrayOf(d.f32), access: \"mutable\" },\n\tpairCount: { uniform: d.u32 },\n});\n\nconst adamicPipeline = (pairIdx: number): void => {\n\t\"use gpu\";\n\tconst u = AdamicLayout.$.pairsU[pairIdx] ?? 0;\n\tconst v = AdamicLayout.$.pairsV[pairIdx] ?? 0;\n\n\tconst uStart = AdamicLayout.$.rowOffsets[u] ?? 0;\n\tconst uEnd = AdamicLayout.$.rowOffsets[u + 1] ?? 0;\n\tconst vStart = AdamicLayout.$.rowOffsets[v] ?? 0;\n\tconst vEnd = AdamicLayout.$.rowOffsets[v + 1] ?? 0;\n\n\tconst degU = uEnd - uStart;\n\tconst degV = vEnd - vStart;\n\n\tAdamicLayout.$.sizeUs[pairIdx] = degU;\n\tAdamicLayout.$.sizeVs[pairIdx] = degV;\n\n\tif (degU === 0 || degV === 0) {\n\t\tAdamicLayout.$.intersections[pairIdx] = 0;\n\t\tAdamicLayout.$.results[pairIdx] = 0.0;\n\t\treturn;\n\t}\n\n\tlet commonCount = 0;\n\tlet sum = 0.0;\n\n\tif (degU <= degV) {\n\t\tfor (let i = uStart; i < uEnd; i = i + 1) {\n\t\t\tconst neighbour = AdamicLayout.$.colIndices[i] ?? 0;\n\t\t\tlet lo = vStart;\n\t\t\tlet hi = vEnd;\n\t\t\twhile (lo < hi) {\n\t\t\t\tconst mid = lo + (hi - lo) / 2;\n\t\t\t\tconst midVal = AdamicLayout.$.colIndices[mid] ?? 0;\n\t\t\t\tif (midVal === neighbour) {\n\t\t\t\t\tcommonCount = commonCount + 1;\n\t\t\t\t\tconst wStart = AdamicLayout.$.rowOffsets[neighbour] ?? 0;\n\t\t\t\t\tconst wEnd = AdamicLayout.$.rowOffsets[neighbour + 1] ?? 0;\n\t\t\t\t\tconst degW = wEnd - wStart;\n\t\t\t\t\tconst invLog = 1.0 / Math.log(degW + 1.0);\n\t\t\t\t\tsum = sum + invLog;\n\t\t\t\t\tlo = hi; // break\n\t\t\t\t} else if (midVal < neighbour) {\n\t\t\t\t\tlo = mid + 1;\n\t\t\t\t} else {\n\t\t\t\t\thi = mid;\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t} else {\n\t\tfor (let i = vStart; i < vEnd; i = i + 1) {\n\t\t\tconst neighbour = AdamicLayout.$.colIndices[i] ?? 0;\n\t\t\tlet lo = uStart;\n\t\t\tlet hi = uEnd;\n\t\t\twhile (lo < hi) {\n\t\t\t\tconst mid = lo + (hi - lo) / 2;\n\t\t\t\tconst midVal = AdamicLayout.$.colIndices[mid] ?? 0;\n\t\t\t\tif (midVal === neighbour) {\n\t\t\t\t\tcommonCount = commonCount + 1;\n\t\t\t\t\tconst wStart = AdamicLayout.$.rowOffsets[neighbour] ?? 0;\n\t\t\t\t\tconst wEnd = AdamicLayout.$.rowOffsets[neighbour + 1] ?? 0;\n\t\t\t\t\tconst degW = wEnd - wStart;\n\t\t\t\t\tconst invLog = 1.0 / Math.log(degW + 1.0);\n\t\t\t\t\tsum = sum + invLog;\n\t\t\t\t\tlo = hi; // break\n\t\t\t\t} else if (midVal < neighbour) {\n\t\t\t\t\tlo = mid + 1;\n\t\t\t\t} else {\n\t\t\t\t\thi = mid;\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\tAdamicLayout.$.intersections[pairIdx] = commonCount;\n\tif (commonCount === 0) {\n\t\tAdamicLayout.$.results[pairIdx] = 0.0;\n\t} else {\n\t\tAdamicLayout.$.results[pairIdx] = (sum * LOG2) / commonCount;\n\t}\n};\n\nexport function dispatchAdamicAdar(\n\troot: GraphwiseGPURoot,\n\tcsrBuffers: TypedBufferGroup,\n\tpairsU: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.u32>>> & StorageFlag,\n\tpairsV: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.u32>>> & StorageFlag,\n\tresults: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.f32>>> & StorageFlag,\n\tintersections: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.u32>>> &\n\t\tStorageFlag,\n\tsizeUs: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.u32>>> & StorageFlag,\n\tsizeVs: TgpuBuffer<ReturnType<typeof d.arrayOf<typeof d.u32>>> & StorageFlag,\n\tpairCount: number,\n): void {\n\tconst pipeline = root.createGuardedComputePipeline(adamicPipeline);\n\tconst pairCountBuffer = root.createBuffer(d.u32, pairCount).$usage(\"uniform\");\n\tconst bindGroup = root.createBindGroup(AdamicLayout, {\n\t\trowOffsets: csrBuffers.rowOffsets,\n\t\tcolIndices: csrBuffers.colIndices,\n\t\tpairsU,\n\t\tpairsV,\n\t\tintersections,\n\t\tsizeUs,\n\t\tsizeVs,\n\t\tresults,\n\t\tpairCount: pairCountBuffer,\n\t});\n\tpipeline.with(bindGroup).dispatchThreads(pairCount);\n}\n\nexport { AdamicLayout };\n"],"mappings":";;;;;;;;;AAYA,IAAM,OAAO;AAEb,IAAM,gBAAA,WAAA,0BAAA,MAAA,IAAe,YAAK,gBAAgB;CACzC,YAAY,EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI,EAAE;CACzC,YAAY,EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI,EAAE;CACzC,QAAQ,EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI,EAAE;CACrC,QAAQ,EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI,EAAE;CACrC,eAAe;EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI;EAAE,QAAQ;EAAW;CAC/D,QAAQ;EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI;EAAE,QAAQ;EAAW;CACxD,QAAQ;EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI;EAAE,QAAQ;EAAW;CACxD,SAAS;EAAE,SAAS,aAAE,QAAQ,aAAE,IAAI;EAAE,QAAQ;EAAW;CACzD,WAAW,EAAE,SAAS,aAAE,KAAA;CACxB,CAAA,EAAA,eAAA;AAED,IAAM,mBAAA,OAAA,WAAA,qCAAA,IAAA,SAAA,EAAA,IAAA,EAAA,MAAkB,YAA0B;AACjD;CACA,MAAM,IAAI,aAAa,EAAE,OAAO,YAAY;CAC5C,MAAM,IAAI,aAAa,EAAE,OAAO,YAAY;CAE5C,MAAM,SAAS,aAAa,EAAE,WAAW,MAAM;CAC/C,MAAM,OAAO,aAAa,EAAE,WAAW,aAAA,GAAA,EAAI,KAAM;CACjD,MAAM,SAAS,aAAa,EAAE,WAAW,MAAM;CAC/C,MAAM,OAAO,aAAa,EAAE,WAAW,aAAA,GAAA,EAAI,KAAM;CAEjD,MAAM,OAAO,aAAA,MAAA,OAAO;CACpB,MAAM,OAAO,aAAA,MAAA,OAAO;AAEpB,cAAa,EAAE,OAAO,WAAW;AACjC,cAAa,EAAE,OAAO,WAAW;AAEjC,KAAI,SAAS,KAAK,SAAS,GAAG;AAC7B,eAAa,EAAE,cAAc,WAAW;AACxC,eAAa,EAAE,QAAQ,WAAW;AAClC;;CAGD,IAAI,cAAc;CAClB,IAAI,MAAM;AAEV,KAAI,QAAQ,KACX,MAAK,IAAI,IAAI,QAAQ,IAAI,MAAM,IAAI,aAAA,GAAA,EAAI,EAAG;EACzC,MAAM,YAAY,aAAa,EAAE,WAAW,MAAM;EAClD,IAAI,KAAK;EACT,IAAI,KAAK;AACT,SAAO,KAAK,IAAI;GACf,MAAM,MAAM,aAAA,IAAA,aAAA,aAAA,IAAA,GAAA,EAAA,EAAA,CAAiB;GAC7B,MAAM,SAAS,aAAa,EAAE,WAAW,QAAQ;AACjD,OAAI,WAAW,WAAW;AACzB,kBAAc,aAAA,aAAA,EAAc;IAC5B,MAAM,SAAS,aAAa,EAAE,WAAW,cAAc;IACvD,MAAM,OAAO,aAAa,EAAE,WAAW,aAAA,WAAA,EAAY,KAAM;IACzD,MAAM,OAAO,aAAA,MAAA,OAAO;IACpB,MAAM,SAAS,aAAA,GAAA,KAAA,IAAA,aAAA,MAAA,EAAA,CAAA,CAA0B;AACzC,UAAM,aAAA,KAAA,OAAM;AACZ,SAAK;cACK,SAAS,UACnB,MAAK,aAAA,KAAA,EAAM;OAEX,MAAK;;;KAKR,MAAK,IAAI,IAAI,QAAQ,IAAI,MAAM,IAAI,aAAA,GAAA,EAAI,EAAG;EACzC,MAAM,YAAY,aAAa,EAAE,WAAW,MAAM;EAClD,IAAI,KAAK;EACT,IAAI,KAAK;AACT,SAAO,KAAK,IAAI;GACf,MAAM,MAAM,aAAA,IAAA,aAAA,aAAA,IAAA,GAAA,EAAA,EAAA,CAAiB;GAC7B,MAAM,SAAS,aAAa,EAAE,WAAW,QAAQ;AACjD,OAAI,WAAW,WAAW;AACzB,kBAAc,aAAA,aAAA,EAAc;IAC5B,MAAM,SAAS,aAAa,EAAE,WAAW,cAAc;IACvD,MAAM,OAAO,aAAa,EAAE,WAAW,aAAA,WAAA,EAAY,KAAM;IACzD,MAAM,OAAO,aAAA,MAAA,OAAO;IACpB,MAAM,SAAS,aAAA,GAAA,KAAA,IAAA,aAAA,MAAA,EAAA,CAAA,CAA0B;AACzC,UAAM,aAAA,KAAA,OAAM;AACZ,SAAK;cACK,SAAS,UACnB,MAAK,aAAA,KAAA,EAAM;OAEX,MAAK;;;AAMT,cAAa,EAAE,cAAc,WAAW;AACxC,KAAI,gBAAgB,EACnB,cAAa,EAAE,QAAQ,WAAW;KAElC,cAAa,EAAE,QAAQ,WAAY,aAAA,aAAA,KAAA,KAAA,EAAA,YAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAInD,SAAgB,mBACf,MACA,YACA,QACA,QACA,SACA,eAEA,QACA,QACA,WACO;CACP,MAAM,YAAA,WAAA,0BAAA,MAAA,IAAW,KAAK,6BAA6B,eAAA,EAAA,WAAA;CACnD,MAAM,mBAAA,WAAA,0BAAA,MAAA,IAAkB,KAAK,aAAa,aAAE,KAAK,UAAU,CAAC,OAAO,UAAA,EAAA,kBAAA;CACnE,MAAM,YAAY,KAAK,gBAAgB,cAAc;EACpD,YAAY,WAAW;EACvB,YAAY,WAAW;EACvB;EACA;EACA;EACA;EACA;EACA;EACA,WAAW;EACX,CAAC;AACF,UAAS,KAAK,UAAU,CAAC,gBAAgB,UAAU"}