graphwise 1.8.0 → 1.9.0

package/dist/gpu/index.cjs

@@ -1,411 +1,16 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-//#region src/gpu/types.ts
-/**
-* Error thrown when GPU backend is requested but unavailable.
-*/
-var GPUNotAvailableError = class extends Error {
-	constructor(reason) {
-		super(`WebGPU not available: ${reason}`);
-		this.name = "GPUNotAvailableError";
-	}
-};
-//#endregion
-//#region src/gpu/csr.ts
-/**
-* Convert a ReadableGraph to CSR format.
-*
-* For undirected graphs, each edge is stored twice (once in each direction).
-* For directed graphs, edges are stored in the out-direction by default.
-*
-* @param graph - The graph to convert
-* @param direction - Edge direction to include (default: 'out' for directed, 'both' for undirected)
-* @returns CSR representation with index mapping
-*/
-function graphToCSR(graph, direction = graph.directed ? "out" : "both") {
-	const nodeToIndex = /* @__PURE__ */ new Map();
-	const indexToNode = [];
-	for (const nodeId of graph.nodeIds()) {
-		const index = indexToNode.length;
-		nodeToIndex.set(nodeId, index);
-		indexToNode.push(nodeId);
-	}
-	const nodeCount = indexToNode.length;
-	const degrees = new Uint32Array(nodeCount);
-	for (const nodeId of graph.nodeIds()) {
-		const srcIndex = nodeToIndex.get(nodeId);
-		if (srcIndex === void 0) continue;
-		degrees[srcIndex] = graph.degree(nodeId, direction);
-	}
-	let totalEdges = 0;
-	for (let i = 0; i < nodeCount; i++) totalEdges += degrees[i] ?? 0;
-	const rowOffsets = new Uint32Array(nodeCount + 1);
-	for (let i = 0; i < nodeCount; i++) rowOffsets[i + 1] = (rowOffsets[i] ?? 0) + (degrees[i] ?? 0);
-	const colIndices = new Uint32Array(totalEdges);
-	const values = new Float32Array(totalEdges);
-	for (const nodeId of graph.nodeIds()) {
-		const srcIndex = nodeToIndex.get(nodeId);
-		if (srcIndex === void 0) continue;
-		const baseOffset = rowOffsets[srcIndex] ?? 0;
-		let localOffset = 0;
-		for (const neighbourId of graph.neighbours(nodeId, direction)) {
-			const dstIndex = nodeToIndex.get(neighbourId);
-			if (dstIndex === void 0) continue;
-			const edgeIdx = baseOffset + localOffset;
-			colIndices[edgeIdx] = dstIndex;
-			values[edgeIdx] = graph.getEdge(nodeId, neighbourId)?.weight ?? 1;
-			localOffset++;
-		}
-	}
-	return {
-		csr: {
-			rowOffsets,
-			colIndices,
-			values,
-			nodeCount,
-			edgeCount: graph.directed ? graph.edgeCount : graph.edgeCount * 2
-		},
-		indexMap: {
-			nodeToIndex,
-			indexToNode
-		}
-	};
-}
-/**
-* Create GPU buffers from a CSR matrix.
-*
-* Buffers are created with:
-* - rowOffsets/colIndices: STORAGE | COPY_DST
-* - values: STORAGE | COPY_DST (if present)
-*
-* @param device - GPU device to create buffers on
-* @param csr - CSR matrix to upload
-* @returns GPU buffer group
-*/
-function csrToGPUBuffers(device, csr) {
-	const rowOffsetsBuffer = device.createBuffer({
-		size: csr.rowOffsets.byteLength,
-		usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST,
-		mappedAtCreation: false
-	});
-	device.queue.writeBuffer(rowOffsetsBuffer, 0, csr.rowOffsets);
-	const colIndicesBuffer = device.createBuffer({
-		size: csr.colIndices.byteLength,
-		usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST,
-		mappedAtCreation: false
-	});
-	device.queue.writeBuffer(colIndicesBuffer, 0, csr.colIndices);
-	let valuesBuffer;
-	if (csr.values !== void 0) {
-		valuesBuffer = device.createBuffer({
-			size: csr.values.byteLength,
-			usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST,
-			mappedAtCreation: false
-		});
-		device.queue.writeBuffer(valuesBuffer, 0, csr.values);
-	}
-	if (valuesBuffer !== void 0) return {
-		rowOffsets: rowOffsetsBuffer,
-		colIndices: colIndicesBuffer,
-		values: valuesBuffer,
-		nodeCount: csr.nodeCount,
-		edgeCount: csr.edgeCount
-	};
-	return {
-		rowOffsets: rowOffsetsBuffer,
-		colIndices: colIndicesBuffer,
-		nodeCount: csr.nodeCount,
-		edgeCount: csr.edgeCount
-	};
-}
-/**
-* Create a result buffer for reading compute output.
-*
-* @param device - GPU device
-* @param byteLength - Size of the buffer in bytes
-* @returns GPU buffer configured for map reading
-*/
-function createResultBuffer(device, byteLength) {
-	return device.createBuffer({
-		size: byteLength,
-		usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.MAP_READ
-	});
-}
-/**
-* Read data from a GPU buffer to CPU.
-*
-* @param device - GPU device
-* @param buffer - Buffer to read from
-* @returns ArrayBuffer containing the buffer data
-*/
-async function readBufferToCPU(device, buffer) {
-	await buffer.mapAsync(GPUMapMode.READ);
-	const data = buffer.getMappedRange().slice(0);
-	buffer.unmap();
-	return data;
-}
-//#endregion
-//#region src/gpu/detect.ts
-/**
-* Detect WebGPU availability in the current environment.
-*
-* Checks for:
-* - Browser: navigator.gpu
-* - Node.js: global GPU constructor (Node.js 21+ with --experimental-webgpu)
-* - Deno: global GPU constructor
-*
-* @returns Detection result with availability status and reason
-*/
-function detectWebGPU() {
-	if (typeof navigator !== "undefined" && "gpu" in navigator) return { available: true };
-	if (typeof globalThis !== "undefined" && "GPU" in globalThis) return { available: true };
-	const reasons = [];
-	if (typeof navigator === "undefined" && typeof globalThis === "undefined") reasons.push("no global scope detected");
-	else if (typeof navigator !== "undefined" && !("gpu" in navigator)) reasons.push("navigator.gpu not present (browser may not support WebGPU)");
-	else if (typeof globalThis !== "undefined" && !("GPU" in globalThis)) reasons.push("global GPU not present (Node.js requires v21+ with --experimental-webgpu flag)");
-	return {
-		available: false,
-		reason: reasons.length > 0 ? reasons.join("; ") : "unknown environment"
-	};
-}
-/**
-* Check if WebGPU is available (convenience function).
-*
-* @returns true if WebGPU is available, false otherwise
-*/
-function isWebGPUAvailable() {
-	return detectWebGPU().available;
-}
-/**
-* Assert that WebGPU is available, throwing an error if not.
-*
-* @throws Error if WebGPU is not available
-*/
-function assertWebGPUAvailable() {
-	const result = detectWebGPU();
-	if (!result.available) throw new Error(`WebGPU required but not available: ${result.reason ?? "unknown reason"}`);
-}
-//#endregion
-//#region src/gpu/context.ts
-/**
-* GPU context management for WebGPU compute operations.
-*
-* Provides device acquisition, buffer pooling, and shader compilation
-* with caching for efficient GPU resource management.
-*/
-/**
-* Type guard to check if an object has a WebGPU-compatible requestAdapter method.
-*/
-function hasRequestAdapter(obj) {
-	if (typeof obj !== "object" || obj === null) return false;
-	if (!("requestAdapter" in obj)) return false;
-	const descriptor = Object.getOwnPropertyDescriptor(obj, "requestAdapter");
-	return descriptor !== void 0 && typeof descriptor.value === "function";
-}
-/**
-* Manages GPU device, buffers, and compiled shaders.
-*
-* Use the singleton pattern via getGPUContext() for most use cases,
-* or create separate contexts for isolated GPU resource pools.
-*/
-var GPUContext = class {
-	device = null;
-	shaderCache = /* @__PURE__ */ new Map();
-	bufferPool = [];
-	destroyedBuffers = /* @__PURE__ */ new WeakSet();
-	/**
-	* Check if this context has an acquired GPU device.
-	*/
-	get isReady() {
-		return this.device !== null;
-	}
-	/**
-	* Get the GPU device, throwing if not acquired.
-	*/
-	getDevice() {
-		if (this.device === null) throw new Error("GPUContext not initialised. Call acquireDevice() first.");
-		return this.device;
-	}
-	/**
-	* Acquire a GPU device from the adapter.
-	*
-	* @param options - Context creation options
-	* @returns true if device was acquired successfully
-	* @throws Error if WebGPU is unavailable or device request fails
-	*/
-	async acquireDevice(options = {}) {
-		const detection = detectWebGPU();
-		if (!detection.available) throw new Error(`WebGPU unavailable: ${detection.reason ?? "unknown reason"}`);
-		let adapter = null;
-		const adapterOpts = options.powerPreference !== void 0 ? { powerPreference: options.powerPreference } : {};
-		if (typeof navigator !== "undefined" && "gpu" in navigator) adapter = await navigator.gpu.requestAdapter(adapterOpts);
-		if (adapter === null && typeof globalThis !== "undefined" && "GPU" in globalThis) {
-			const gpuDescriptor = Object.getOwnPropertyDescriptor(globalThis, "gpu");
-			if (gpuDescriptor !== void 0 && hasRequestAdapter(gpuDescriptor.value)) adapter = await gpuDescriptor.value.requestAdapter(adapterOpts);
-		}
-		if (adapter === null) throw new Error("No GPU adapter found");
-		this.device = await adapter.requestDevice();
-		this.device.lost.then((info) => {
-			console.error(`GPU device lost: ${info.message}`);
-			this.device = null;
-			this.clearCache();
-		});
-		return true;
-	}
-	/**
-	* Compile a WGSL shader, using cache if available.
-	*
-	* @param code - WGSL shader code
-	* @param key - Optional cache key (defaults to code hash)
-	* @returns Compiled shader module
-	*/
-	compileShader(code, key) {
-		const cacheKey = key ?? this.hashCode(code);
-		const cached = this.shaderCache.get(cacheKey);
-		if (cached?.code === code) return cached.module;
-		const module = this.getDevice().createShaderModule({ code });
-		this.shaderCache.set(cacheKey, {
-			module,
-			code
-		});
-		return module;
-	}
-	/**
-	* Create a GPU buffer, optionally reusing from pool.
-	*
-	* @param size - Buffer size in bytes
-	* @param usage - Buffer usage flags
-	* @returns GPU buffer
-	*/
-	createBuffer(size, usage) {
-		for (let i = 0; i < this.bufferPool.length; i++) {
-			const buffer = this.bufferPool[i];
-			if (buffer !== void 0 && !this.destroyedBuffers.has(buffer) && buffer.size >= size && (buffer.usage & usage) === usage) {
-				this.bufferPool.splice(i, 1);
-				return buffer;
-			}
-		}
-		return this.getDevice().createBuffer({
-			size,
-			usage
-		});
-	}
-	/**
-	* Return a buffer to the pool for reuse.
-	*
-	* @param buffer - Buffer to recycle
-	*/
-	recycleBuffer(buffer) {
-		if (!this.destroyedBuffers.has(buffer)) this.bufferPool.push(buffer);
-	}
-	/**
-	* Mark a buffer as destroyed (call before GPUBuffer.destroy()).
-	*
-	* @param buffer - Buffer to mark as destroyed
-	*/
-	markDestroyed(buffer) {
-		this.destroyedBuffers.add(buffer);
-	}
-	/**
-	* Create a compute pipeline from shader code.
-	*
-	* @param code - WGSL compute shader code
-	* @param entryPoint - Entry point function name
-	* @returns Compute pipeline
-	*/
-	createComputePipeline(code, entryPoint = "main") {
-		const device = this.getDevice();
-		const module = this.compileShader(code);
-		return device.createComputePipeline({
-			layout: "auto",
-			compute: {
-				module,
-				entryPoint
-			}
-		});
-	}
-	/**
-	* Create a compute pipeline asynchronously (preferred for performance).
-	*
-	* @param code - WGSL compute shader code
-	* @param entryPoint - Entry point function name
-	* @returns Promise resolving to compute pipeline
-	*/
-	async createComputePipelineAsync(code, entryPoint = "main") {
-		const device = this.getDevice();
-		const module = this.compileShader(code);
-		return device.createComputePipelineAsync({
-			layout: "auto",
-			compute: {
-				module,
-				entryPoint
-			}
-		});
-	}
-	/**
-	* Clear all cached resources.
-	*/
-	clearCache() {
-		this.shaderCache.clear();
-		for (const buffer of this.bufferPool) buffer.destroy();
-		this.bufferPool.length = 0;
-	}
-	/**
-	* Destroy the context and release all resources.
-	*/
-	destroy() {
-		this.clearCache();
-		if (this.device !== null) {
-			this.device.destroy();
-			this.device = null;
-		}
-	}
-	/**
-	* Simple string hash for cache keys.
-	*/
-	hashCode(str) {
-		let hash = 0;
-		for (let i = 0; i < str.length; i++) {
-			const char = str.charCodeAt(i);
-			hash = (hash << 5) - hash + char | 0;
-		}
-		return hash.toString(16);
-	}
-};
-var defaultContext = null;
-/**
-* Get or create the default GPU context.
-*
-* This is a lazy singleton - the device is not acquired until
-* acquireDevice() is called.
-*
-* @param options - Context creation options
-* @returns GPU context instance
-*/
-function getGPUContext(options) {
-	if (defaultContext === null || (options?.forceNew ?? false)) defaultContext = new GPUContext();
-	return defaultContext;
-}
-/**
-* Create a new isolated GPU context.
-*
-* Use this when you need separate resource pools or device management.
-*
-* @returns New GPU context instance
-*/
-function createGPUContext() {
-	return new GPUContext();
-}
-//#endregion
-exports.GPUContext = GPUContext;
-exports.GPUNotAvailableError = GPUNotAvailableError;
-exports.assertWebGPUAvailable = assertWebGPUAvailable;
-exports.createGPUContext = createGPUContext;
-exports.createResultBuffer = createResultBuffer;
-exports.csrToGPUBuffers = csrToGPUBuffers;
-exports.detectWebGPU = detectWebGPU;
-exports.getGPUContext = getGPUContext;
-exports.graphToCSR = graphToCSR;
-exports.isWebGPUAvailable = isWebGPUAvailable;
-exports.readBufferToCPU = readBufferToCPU;
-
-//# sourceMappingURL=index.cjs.map
+const require_gpu = require("../gpu-Y6owRVMi.cjs");
+exports.GPUNotAvailableError = require_gpu.GPUNotAvailableError;
+exports.assertWebGPUAvailable = require_gpu.assertWebGPUAvailable;
+exports.csrToTypedBuffers = require_gpu.csrToTypedBuffers;
+exports.detectWebGPU = require_gpu.detectWebGPU;
+exports.gpuBfsLevels = require_gpu.gpuBfsLevels;
+exports.gpuDegreeHistogram = require_gpu.gpuDegreeHistogram;
+exports.gpuJaccardBatch = require_gpu.gpuJaccardBatch;
+exports.gpuPageRank = require_gpu.gpuPageRank;
+exports.gpuSpmv = require_gpu.gpuSpmv;
+exports.graphToCSR = require_gpu.graphToCSR;
+exports.initGPU = require_gpu.initGPU;
+exports.initGPUFromDevice = require_gpu.initGPUFromDevice;
+exports.isWebGPUAvailable = require_gpu.isWebGPUAvailable;
+exports.withBackend = require_gpu.withBackend;

package/dist/gpu/index.d.ts

@@ -6,5 +6,7 @@
 export * from './types';
 export * from './csr';
 export * from './detect';
-export * from './context';
+export * from './root';
+export * from './dispatch';
+export * from './operations';
 //# sourceMappingURL=index.d.ts.map

package/dist/gpu/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/gpu/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,UAAU,CAAC;AACzB,cAAc,WAAW,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/gpu/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,UAAU,CAAC;AACzB,cAAc,QAAQ,CAAC;AACvB,cAAc,YAAY,CAAC;AAC3B,cAAc,cAAc,CAAC"}