graphwise 1.9.0 → 1.10.0

package/dist/kernel-2oH4Cn32.cjs

@@ -0,0 +1,1001 @@
+const require_typegpu = require("./typegpu-Dq5FfUB8.cjs");
+//#region src/gpu/kernels/adamic-adar/kernel.ts
+/**
+* TypeGPU compute kernel for Adamic-Adar index (batch).
+*
+* For each pair (u, v): iterate smaller neighbourhood, detect common neighbours
+* by binary search in the other neighbourhood, sum 1 / log(deg(w) + 1) and
+* normalise by log(2)/count to match CPU normalisation.
+*/
+var LOG2 = .6931471805599453;
+var AdamicLayout = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(require_typegpu.src_default.bindGroupLayout({
+	rowOffsets: { storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32) },
+	colIndices: { storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32) },
+	pairsU: { storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32) },
+	pairsV: { storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32) },
+	intersections: {
+		storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32),
+		access: "mutable"
+	},
+	sizeUs: {
+		storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32),
+		access: "mutable"
+	},
+	sizeVs: {
+		storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.u32),
+		access: "mutable"
+	},
+	results: {
+		storage: require_typegpu.data_exports.arrayOf(require_typegpu.data_exports.f32),
+		access: "mutable"
+	},
+	pairCount: { uniform: require_typegpu.data_exports.u32 }
+}), "AdamicLayout");
+var adamicPipeline = (($) => (globalThis.__TYPEGPU_META__ ??= /* @__PURE__ */ new WeakMap()).set($.f = ((pairIdx) => {
+	"use gpu";
+	const u = AdamicLayout.$.pairsU[pairIdx] ?? 0;
+	const v = AdamicLayout.$.pairsV[pairIdx] ?? 0;
+	const uStart = AdamicLayout.$.rowOffsets[u] ?? 0;
+	const uEnd = AdamicLayout.$.rowOffsets[__tsover_add(u, 1)] ?? 0;
+	const vStart = AdamicLayout.$.rowOffsets[v] ?? 0;
+	const vEnd = AdamicLayout.$.rowOffsets[__tsover_add(v, 1)] ?? 0;
+	const degU = __tsover_sub(uEnd, uStart);
+	const degV = __tsover_sub(vEnd, vStart);
+	AdamicLayout.$.sizeUs[pairIdx] = degU;
+	AdamicLayout.$.sizeVs[pairIdx] = degV;
+	if (degU === 0 || degV === 0) {
+		AdamicLayout.$.intersections[pairIdx] = 0;
+		AdamicLayout.$.results[pairIdx] = 0;
+		return;
+	}
+	let commonCount = 0;
+	let sum = 0;
+	if (degU <= degV) for (let i = uStart; i < uEnd; i = __tsover_add(i, 1)) {
+		const neighbour = AdamicLayout.$.colIndices[i] ?? 0;
+		let lo = vStart;
+		let hi = vEnd;
+		while (lo < hi) {
+			const mid = __tsover_add(lo, __tsover_div(__tsover_sub(hi, lo), 2));
+			const midVal = AdamicLayout.$.colIndices[mid] ?? 0;
+			if (midVal === neighbour) {
+				commonCount = __tsover_add(commonCount, 1);
+				const wStart = AdamicLayout.$.rowOffsets[neighbour] ?? 0;
+				const wEnd = AdamicLayout.$.rowOffsets[__tsover_add(neighbour, 1)] ?? 0;
+				const degW = __tsover_sub(wEnd, wStart);
+				const invLog = __tsover_div(1, Math.log(__tsover_add(degW, 1)));
+				sum = __tsover_add(sum, invLog);
+				lo = hi;
+			} else if (midVal < neighbour) lo = __tsover_add(mid, 1);
+			else hi = mid;
+		}
+	}
+	else for (let i = vStart; i < vEnd; i = __tsover_add(i, 1)) {
+		const neighbour = AdamicLayout.$.colIndices[i] ?? 0;
+		let lo = uStart;
+		let hi = uEnd;
+		while (lo < hi) {
+			const mid = __tsover_add(lo, __tsover_div(__tsover_sub(hi, lo), 2));
+			const midVal = AdamicLayout.$.colIndices[mid] ?? 0;
+			if (midVal === neighbour) {
+				commonCount = __tsover_add(commonCount, 1);
+				const wStart = AdamicLayout.$.rowOffsets[neighbour] ?? 0;
+				const wEnd = AdamicLayout.$.rowOffsets[__tsover_add(neighbour, 1)] ?? 0;
+				const degW = __tsover_sub(wEnd, wStart);
+				const invLog = __tsover_div(1, Math.log(__tsover_add(degW, 1)));
+				sum = __tsover_add(sum, invLog);
+				lo = hi;
+			} else if (midVal < neighbour) lo = __tsover_add(mid, 1);
+			else hi = mid;
+		}
+	}
+	AdamicLayout.$.intersections[pairIdx] = commonCount;
+	if (commonCount === 0) AdamicLayout.$.results[pairIdx] = 0;
+	else AdamicLayout.$.results[pairIdx] = __tsover_div(__tsover_mul(sum, LOG2), commonCount);
+}), {
+	v: 1,
+	name: "adamicPipeline",
+	ast: {
+		"params": [{
+			"type": "i",
+			"name": "pairIdx"
+		}],
+		"body": [0, [
+			[
+				13,
+				"u",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"pairsU"
+						],
+						"pairIdx"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"v",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"pairsV"
+						],
+						"pairIdx"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"uStart",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						"u"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"uEnd",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						[
+							1,
+							"u",
+							"+",
+							[5, "1"]
+						]
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"vStart",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						"v"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"vEnd",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						[
+							1,
+							"v",
+							"+",
+							[5, "1"]
+						]
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"degU",
+				[
+					1,
+					"uEnd",
+					"-",
+					"uStart"
+				]
+			],
+			[
+				13,
+				"degV",
+				[
+					1,
+					"vEnd",
+					"-",
+					"vStart"
+				]
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"AdamicLayout",
+							"$"
+						],
+						"sizeUs"
+					],
+					"pairIdx"
+				],
+				"=",
+				"degU"
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"AdamicLayout",
+							"$"
+						],
+						"sizeVs"
+					],
+					"pairIdx"
+				],
+				"=",
+				"degV"
+			],
+			[
+				11,
+				[
+					3,
+					[
+						1,
+						"degU",
+						"===",
+						[5, "0"]
+					],
+					"||",
+					[
+						1,
+						"degV",
+						"===",
+						[5, "0"]
+					]
+				],
+				[0, [
+					[
+						2,
+						[
+							8,
+							[
+								7,
+								[
+									7,
+									"AdamicLayout",
+									"$"
+								],
+								"intersections"
+							],
+							"pairIdx"
+						],
+						"=",
+						[5, "0"]
+					],
+					[
+						2,
+						[
+							8,
+							[
+								7,
+								[
+									7,
+									"AdamicLayout",
+									"$"
+								],
+								"results"
+							],
+							"pairIdx"
+						],
+						"=",
+						[5, "0"]
+					],
+					[10]
+				]]
+			],
+			[
+				12,
+				"commonCount",
+				[5, "0"]
+			],
+			[
+				12,
+				"sum",
+				[5, "0"]
+			],
+			[
+				11,
+				[
+					1,
+					"degU",
+					"<=",
+					"degV"
+				],
+				[0, [[
+					14,
+					[
+						12,
+						"i",
+						"uStart"
+					],
+					[
+						1,
+						"i",
+						"<",
+						"uEnd"
+					],
+					[
+						2,
+						"i",
+						"=",
+						[
+							1,
+							"i",
+							"+",
+							[5, "1"]
+						]
+					],
+					[0, [
+						[
+							13,
+							"neighbour",
+							[
+								3,
+								[
+									8,
+									[
+										7,
+										[
+											7,
+											"AdamicLayout",
+											"$"
+										],
+										"colIndices"
+									],
+									"i"
+								],
+								"??",
+								[5, "0"]
+							]
+						],
+						[
+							12,
+							"lo",
+							"vStart"
+						],
+						[
+							12,
+							"hi",
+							"vEnd"
+						],
+						[
+							15,
+							[
+								1,
+								"lo",
+								"<",
+								"hi"
+							],
+							[0, [
+								[
+									13,
+									"mid",
+									[
+										1,
+										"lo",
+										"+",
+										[
+											1,
+											[
+												1,
+												"hi",
+												"-",
+												"lo"
+											],
+											"/",
+											[5, "2"]
+										]
+									]
+								],
+								[
+									13,
+									"midVal",
+									[
+										3,
+										[
+											8,
+											[
+												7,
+												[
+													7,
+													"AdamicLayout",
+													"$"
+												],
+												"colIndices"
+											],
+											"mid"
+										],
+										"??",
+										[5, "0"]
+									]
+								],
+								[
+									11,
+									[
+										1,
+										"midVal",
+										"===",
+										"neighbour"
+									],
+									[0, [
+										[
+											2,
+											"commonCount",
+											"=",
+											[
+												1,
+												"commonCount",
+												"+",
+												[5, "1"]
+											]
+										],
+										[
+											13,
+											"wStart",
+											[
+												3,
+												[
+													8,
+													[
+														7,
+														[
+															7,
+															"AdamicLayout",
+															"$"
+														],
+														"rowOffsets"
+													],
+													"neighbour"
+												],
+												"??",
+												[5, "0"]
+											]
+										],
+										[
+											13,
+											"wEnd",
+											[
+												3,
+												[
+													8,
+													[
+														7,
+														[
+															7,
+															"AdamicLayout",
+															"$"
+														],
+														"rowOffsets"
+													],
+													[
+														1,
+														"neighbour",
+														"+",
+														[5, "1"]
+													]
+												],
+												"??",
+												[5, "0"]
+											]
+										],
+										[
+											13,
+											"degW",
+											[
+												1,
+												"wEnd",
+												"-",
+												"wStart"
+											]
+										],
+										[
+											13,
+											"invLog",
+											[
+												1,
+												[5, "1"],
+												"/",
+												[
+													6,
+													[
+														7,
+														"Math",
+														"log"
+													],
+													[[
+														1,
+														"degW",
+														"+",
+														[5, "1"]
+													]]
+												]
+											]
+										],
+										[
+											2,
+											"sum",
+											"=",
+											[
+												1,
+												"sum",
+												"+",
+												"invLog"
+											]
+										],
+										[
+											2,
+											"lo",
+											"=",
+											"hi"
+										]
+									]],
+									[
+										11,
+										[
+											1,
+											"midVal",
+											"<",
+											"neighbour"
+										],
+										[0, [[
+											2,
+											"lo",
+											"=",
+											[
+												1,
+												"mid",
+												"+",
+												[5, "1"]
+											]
+										]]],
+										[0, [[
+											2,
+											"hi",
+											"=",
+											"mid"
+										]]]
+									]
+								]
+							]]
+						]
+					]]
+				]]],
+				[0, [[
+					14,
+					[
+						12,
+						"i",
+						"vStart"
+					],
+					[
+						1,
+						"i",
+						"<",
+						"vEnd"
+					],
+					[
+						2,
+						"i",
+						"=",
+						[
+							1,
+							"i",
+							"+",
+							[5, "1"]
+						]
+					],
+					[0, [
+						[
+							13,
+							"neighbour",
+							[
+								3,
+								[
+									8,
+									[
+										7,
+										[
+											7,
+											"AdamicLayout",
+											"$"
+										],
+										"colIndices"
+									],
+									"i"
+								],
+								"??",
+								[5, "0"]
+							]
+						],
+						[
+							12,
+							"lo",
+							"uStart"
+						],
+						[
+							12,
+							"hi",
+							"uEnd"
+						],
+						[
+							15,
+							[
+								1,
+								"lo",
+								"<",
+								"hi"
+							],
+							[0, [
+								[
+									13,
+									"mid",
+									[
+										1,
+										"lo",
+										"+",
+										[
+											1,
+											[
+												1,
+												"hi",
+												"-",
+												"lo"
+											],
+											"/",
+											[5, "2"]
+										]
+									]
+								],
+								[
+									13,
+									"midVal",
+									[
+										3,
+										[
+											8,
+											[
+												7,
+												[
+													7,
+													"AdamicLayout",
+													"$"
+												],
+												"colIndices"
+											],
+											"mid"
+										],
+										"??",
+										[5, "0"]
+									]
+								],
+								[
+									11,
+									[
+										1,
+										"midVal",
+										"===",
+										"neighbour"
+									],
+									[0, [
+										[
+											2,
+											"commonCount",
+											"=",
+											[
+												1,
+												"commonCount",
+												"+",
+												[5, "1"]
+											]
+										],
+										[
+											13,
+											"wStart",
+											[
+												3,
+												[
+													8,
+													[
+														7,
+														[
+															7,
+															"AdamicLayout",
+															"$"
+														],
+														"rowOffsets"
+													],
+													"neighbour"
+												],
+												"??",
+												[5, "0"]
+											]
+										],
+										[
+											13,
+											"wEnd",
+											[
+												3,
+												[
+													8,
+													[
+														7,
+														[
+															7,
+															"AdamicLayout",
+															"$"
+														],
+														"rowOffsets"
+													],
+													[
+														1,
+														"neighbour",
+														"+",
+														[5, "1"]
+													]
+												],
+												"??",
+												[5, "0"]
+											]
+										],
+										[
+											13,
+											"degW",
+											[
+												1,
+												"wEnd",
+												"-",
+												"wStart"
+											]
+										],
+										[
+											13,
+											"invLog",
+											[
+												1,
+												[5, "1"],
+												"/",
+												[
+													6,
+													[
+														7,
+														"Math",
+														"log"
+													],
+													[[
+														1,
+														"degW",
+														"+",
+														[5, "1"]
+													]]
+												]
+											]
+										],
+										[
+											2,
+											"sum",
+											"=",
+											[
+												1,
+												"sum",
+												"+",
+												"invLog"
+											]
+										],
+										[
+											2,
+											"lo",
+											"=",
+											"hi"
+										]
+									]],
+									[
+										11,
+										[
+											1,
+											"midVal",
+											"<",
+											"neighbour"
+										],
+										[0, [[
+											2,
+											"lo",
+											"=",
+											[
+												1,
+												"mid",
+												"+",
+												[5, "1"]
+											]
+										]]],
+										[0, [[
+											2,
+											"hi",
+											"=",
+											"mid"
+										]]]
+									]
+								]
+							]]
+						]
+					]]
+				]]]
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"AdamicLayout",
+							"$"
+						],
+						"intersections"
+					],
+					"pairIdx"
+				],
+				"=",
+				"commonCount"
+			],
+			[
+				11,
+				[
+					1,
+					"commonCount",
+					"===",
+					[5, "0"]
+				],
+				[0, [[
+					2,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"results"
+						],
+						"pairIdx"
+					],
+					"=",
+					[5, "0"]
+				]]],
+				[0, [[
+					2,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"AdamicLayout",
+								"$"
+							],
+							"results"
+						],
+						"pairIdx"
+					],
+					"=",
+					[
+						1,
+						[
+							1,
+							"sum",
+							"*",
+							"LOG2"
+						],
+						"/",
+						"commonCount"
+					]
+				]]]
+			]
+		]],
+		"externalNames": [
+			"AdamicLayout",
+			"Math",
+			"LOG2"
+		]
+	},
+	externals: () => ({
+		AdamicLayout,
+		Math,
+		LOG2
+	})
+}) && $.f)({});
+function dispatchAdamicAdar(root, csrBuffers, pairsU, pairsV, results, intersections, sizeUs, sizeVs, pairCount) {
+	const pipeline = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(root.createGuardedComputePipeline(adamicPipeline), "pipeline");
+	const pairCountBuffer = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(root.createBuffer(require_typegpu.data_exports.u32, pairCount).$usage("uniform"), "pairCountBuffer");
+	const bindGroup = root.createBindGroup(AdamicLayout, {
+		rowOffsets: csrBuffers.rowOffsets,
+		colIndices: csrBuffers.colIndices,
+		pairsU,
+		pairsV,
+		intersections,
+		sizeUs,
+		sizeVs,
+		results,
+		pairCount: pairCountBuffer
+	});
+	pipeline.with(bindGroup).dispatchThreads(pairCount);
+}
+//#endregion
+exports.dispatchAdamicAdar = dispatchAdamicAdar;
+
+//# sourceMappingURL=kernel-2oH4Cn32.cjs.map