graphwise 1.9.1 → 1.11.0

package/dist/kernel-6deK9fh1.js

@@ -0,0 +1,724 @@
+import { n as data_exports, t as src_default } from "./typegpu-DwnJf28i.js";
+//#region src/gpu/kernels/intersection/kernel.ts
+/**
+* TypeGPU compute kernel for batch neighbourhood intersection.
+*
+* Computes intersection size and neighbourhood sizes for multiple node pairs.
+* One thread per pair: iterates the smaller neighbourhood and binary-searches the other.
+*
+* @module gpu/kernels/intersection/kernel
+*/
+/**
+* Bind group layout for Intersection kernel.
+*/
+var IntersectionLayout = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(src_default.bindGroupLayout({
+	rowOffsets: { storage: data_exports.arrayOf(data_exports.u32) },
+	colIndices: { storage: data_exports.arrayOf(data_exports.u32) },
+	pairsU: { storage: data_exports.arrayOf(data_exports.u32) },
+	pairsV: { storage: data_exports.arrayOf(data_exports.u32) },
+	intersections: {
+		storage: data_exports.arrayOf(data_exports.u32),
+		access: "mutable"
+	},
+	sizeUs: {
+		storage: data_exports.arrayOf(data_exports.u32),
+		access: "mutable"
+	},
+	sizeVs: {
+		storage: data_exports.arrayOf(data_exports.u32),
+		access: "mutable"
+	},
+	pairCount: { uniform: data_exports.u32 }
+}), "IntersectionLayout");
+/**
+* Intersection compute pipeline: one thread per pair.
+*
+* For each pair (u, v):
+*  - Count intersection by iterating smaller neighbourhood
+*  - Binary search in larger neighbourhood
+*  - Write intersection, sizeU, sizeV to output buffers
+*/
+var intersectionPipeline = (($) => (globalThis.__TYPEGPU_META__ ??= /* @__PURE__ */ new WeakMap()).set($.f = ((pairIdx) => {
+	"use gpu";
+	const u = IntersectionLayout.$.pairsU[pairIdx] ?? 0;
+	const v = IntersectionLayout.$.pairsV[pairIdx] ?? 0;
+	const uStart = IntersectionLayout.$.rowOffsets[u] ?? 0;
+	const uEnd = IntersectionLayout.$.rowOffsets[__tsover_add(u, 1)] ?? 0;
+	const vStart = IntersectionLayout.$.rowOffsets[v] ?? 0;
+	const vEnd = IntersectionLayout.$.rowOffsets[__tsover_add(v, 1)] ?? 0;
+	const degU = __tsover_sub(uEnd, uStart);
+	const degV = __tsover_sub(vEnd, vStart);
+	IntersectionLayout.$.sizeUs[pairIdx] = degU;
+	IntersectionLayout.$.sizeVs[pairIdx] = degV;
+	if (degU === 0 || degV === 0) {
+		IntersectionLayout.$.intersections[pairIdx] = 0;
+		return;
+	}
+	let intersection = 0;
+	if (degU <= degV) for (let i = uStart; i < uEnd; i = __tsover_add(i, 1)) {
+		const neighbour = IntersectionLayout.$.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 = IntersectionLayout.$.colIndices[mid] ?? 0;
+			if (midVal === neighbour) {
+				intersection = __tsover_add(intersection, 1);
+				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 = IntersectionLayout.$.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 = IntersectionLayout.$.colIndices[mid] ?? 0;
+			if (midVal === neighbour) {
+				intersection = __tsover_add(intersection, 1);
+				lo = hi;
+			} else if (midVal < neighbour) lo = __tsover_add(mid, 1);
+			else hi = mid;
+		}
+	}
+	IntersectionLayout.$.intersections[pairIdx] = intersection;
+}), {
+	v: 1,
+	name: "intersectionPipeline",
+	ast: {
+		"params": [{
+			"type": "i",
+			"name": "pairIdx"
+		}],
+		"body": [0, [
+			[
+				13,
+				"u",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"pairsU"
+						],
+						"pairIdx"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"v",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"pairsV"
+						],
+						"pairIdx"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"uStart",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						"u"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"uEnd",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						[
+							1,
+							"u",
+							"+",
+							[5, "1"]
+						]
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"vStart",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						"v"
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"vEnd",
+				[
+					3,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"rowOffsets"
+						],
+						[
+							1,
+							"v",
+							"+",
+							[5, "1"]
+						]
+					],
+					"??",
+					[5, "0"]
+				]
+			],
+			[
+				13,
+				"degU",
+				[
+					1,
+					"uEnd",
+					"-",
+					"uStart"
+				]
+			],
+			[
+				13,
+				"degV",
+				[
+					1,
+					"vEnd",
+					"-",
+					"vStart"
+				]
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"IntersectionLayout",
+							"$"
+						],
+						"sizeUs"
+					],
+					"pairIdx"
+				],
+				"=",
+				"degU"
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"IntersectionLayout",
+							"$"
+						],
+						"sizeVs"
+					],
+					"pairIdx"
+				],
+				"=",
+				"degV"
+			],
+			[
+				11,
+				[
+					3,
+					[
+						1,
+						"degU",
+						"===",
+						[5, "0"]
+					],
+					"||",
+					[
+						1,
+						"degV",
+						"===",
+						[5, "0"]
+					]
+				],
+				[0, [[
+					2,
+					[
+						8,
+						[
+							7,
+							[
+								7,
+								"IntersectionLayout",
+								"$"
+							],
+							"intersections"
+						],
+						"pairIdx"
+					],
+					"=",
+					[5, "0"]
+				], [10]]]
+			],
+			[
+				12,
+				"intersection",
+				[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,
+											"IntersectionLayout",
+											"$"
+										],
+										"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,
+													"IntersectionLayout",
+													"$"
+												],
+												"colIndices"
+											],
+											"mid"
+										],
+										"??",
+										[5, "0"]
+									]
+								],
+								[
+									11,
+									[
+										1,
+										"midVal",
+										"===",
+										"neighbour"
+									],
+									[0, [[
+										2,
+										"intersection",
+										"=",
+										[
+											1,
+											"intersection",
+											"+",
+											[5, "1"]
+										]
+									], [
+										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,
+											"IntersectionLayout",
+											"$"
+										],
+										"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,
+													"IntersectionLayout",
+													"$"
+												],
+												"colIndices"
+											],
+											"mid"
+										],
+										"??",
+										[5, "0"]
+									]
+								],
+								[
+									11,
+									[
+										1,
+										"midVal",
+										"===",
+										"neighbour"
+									],
+									[0, [[
+										2,
+										"intersection",
+										"=",
+										[
+											1,
+											"intersection",
+											"+",
+											[5, "1"]
+										]
+									], [
+										2,
+										"lo",
+										"=",
+										"hi"
+									]]],
+									[
+										11,
+										[
+											1,
+											"midVal",
+											"<",
+											"neighbour"
+										],
+										[0, [[
+											2,
+											"lo",
+											"=",
+											[
+												1,
+												"mid",
+												"+",
+												[5, "1"]
+											]
+										]]],
+										[0, [[
+											2,
+											"hi",
+											"=",
+											"mid"
+										]]]
+									]
+								]
+							]]
+						]
+					]]
+				]]]
+			],
+			[
+				2,
+				[
+					8,
+					[
+						7,
+						[
+							7,
+							"IntersectionLayout",
+							"$"
+						],
+						"intersections"
+					],
+					"pairIdx"
+				],
+				"=",
+				"intersection"
+			]
+		]],
+		"externalNames": ["IntersectionLayout"]
+	},
+	externals: () => ({ IntersectionLayout })
+}) && $.f)({});
+/**
+* Dispatch batch intersection on GPU.
+*
+* @param root - TypeGPU root instance
+* @param csrBuffers - CSR matrix as typed buffers
+* @param pairsU - First node of each pair (u32 array)
+* @param pairsV - Second node of each pair (u32 array)
+* @param intersections - Output intersections (u32 array, mutable)
+* @param sizeUs - Output sizeU per pair (u32 array, mutable)
+* @param sizeVs - Output sizeV per pair (u32 array, mutable)
+* @param pairCount - Number of pairs to compute
+*/
+function dispatchIntersection(root, csrBuffers, pairsU, pairsV, intersections, sizeUs, sizeVs, pairCount) {
+	const pipeline = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(root.createGuardedComputePipeline(intersectionPipeline), "pipeline");
+	const pairCountBuffer = (globalThis.__TYPEGPU_AUTONAME__ ?? ((a) => a))(root.createBuffer(data_exports.u32, pairCount).$usage("uniform"), "pairCountBuffer");
+	const bindGroup = root.createBindGroup(IntersectionLayout, {
+		rowOffsets: csrBuffers.rowOffsets,
+		colIndices: csrBuffers.colIndices,
+		pairsU,
+		pairsV,
+		intersections,
+		sizeUs,
+		sizeVs,
+		pairCount: pairCountBuffer
+	});
+	pipeline.with(bindGroup).dispatchThreads(pairCount);
+}
+//#endregion
+export { dispatchIntersection };
+
+//# sourceMappingURL=kernel-6deK9fh1.js.map