From: Chris Duncan Date: Sun, 12 Jan 2025 07:28:15 +0000 (-0800) Subject: Replace validate shared code with dispatch call. X-Git-Tag: v1.1.0~5 X-Git-Url: https://zoso.dev/?a=commitdiff_plain;h=c1a105e2a2eb4604091e9808110e43635cb1d2be;p=nano-pow.git Replace validate shared code with dispatch call. --- diff --git a/src/classes/gpu.ts b/src/classes/gpu.ts index 42659cc..3e7c325 100644 --- a/src/classes/gpu.ts +++ b/src/classes/gpu.ts @@ -225,68 +225,7 @@ export class NanoPowGpu { } if (this.#device == null) throw new Error(`WebGPU device failed to load.`) - // Set up uniform buffer object - // Note: u32 size is 4, but total alignment must be multiple of 16 - const uboView = new DataView(new ArrayBuffer(48)) - for (let i = 0; i < 64; i += 8) { - const uint32 = hash.slice(i, i + 8) - uboView.setUint32(i / 2, parseInt(uint32, 16)) - } - uboView.setBigUint64(32, BigInt(`0x${work}`), true) - uboView.setUint32(40, threshold, true) - this.#device.queue.writeBuffer(this.#uboBuffer, 0, uboView) - - // Reset `nonce` and `found` to 0u in WORK before each calculation - this.#device.queue.writeBuffer(this.#gpuBuffer, 0, new Uint32Array([0, 0, 0])) - - // Bind UBO read and GPU write buffers - const bindGroup = this.#device.createBindGroup({ - layout: this.#bindGroupLayout, - entries: [ - { - binding: 0, - resource: { - buffer: this.#uboBuffer - }, - }, - { - binding: 1, - resource: { - buffer: this.#gpuBuffer - }, - }, - ], - }) - - // Create command encoder to issue commands to GPU and initiate computation - const commandEncoder = this.#device.createCommandEncoder() - const passEncoder = commandEncoder.beginComputePass() - - // Issue commands and end compute pass structure - passEncoder.setPipeline(this.#pipeline) - passEncoder.setBindGroup(0, bindGroup) - passEncoder.dispatchWorkgroups(1) - passEncoder.end() - - // Copy 8-byte nonce and 4-byte found flag from GPU to CPU for reading - commandEncoder.copyBufferToBuffer(this.#gpuBuffer, 0, this.#cpuBuffer, 0, 12) - - // End computation by passing array of command buffers to command queue for execution - this.#device.queue.submit([commandEncoder.finish()]) - - // Read results back to Javascript and then unmap buffer after reading - let data = null - try { - await this.#cpuBuffer.mapAsync(GPUMapMode.READ) - await this.#device.queue.onSubmittedWorkDone() - data = new DataView(this.#cpuBuffer.getMappedRange().slice(0)) - this.#cpuBuffer.unmap() - } catch (err) { - console.warn(`Error getting data from GPU. ${err}`) - return this.validate(work, hash, threshold) - } - if (data == null) throw new Error(`Failed to get data from buffer.`) - + const data = await this.#dispatch(BigInt(`0x${work}`), hash, threshold, 1) const nonce = data.getBigUint64(0, true).toString(16).padStart(16, '0') const found = !!data.getUint32(8) this.#busy = false diff --git a/test.html b/test.html index 9cc20dc..ffe3bca 100644 --- a/test.html +++ b/test.html @@ -9,7 +9,7 @@ SPDX-License-Identifier: GPL-3.0-or-later - +