}
// Initialize WebGPU
- static async init () {
+ static async init (): Promise<void> {
if (this.#busy) return
this.#busy = true
// Request device and adapter
if (!(device instanceof GPUDevice)) {
throw new Error('WebGPU device failed to load.')
}
- device.lost.then(loss => {
- console.dir(loss)
- console.warn(`Device lost. Reinitializing...`)
- this.#cpuBuffer?.destroy()
- this.#gpuBuffer?.destroy()
- this.#uboBuffer?.destroy()
- this.init()
- })
+ device.lost.then(this.reset)
this.#device = device
this.setup()
} catch (err) {
}
}
- static setup () {
+ static setup (): void {
if (this.#device == null) throw new Error(`WebGPU device failed to load.`)
- // Create buffers for writing GPU calculations and reading from Javascript
+ // Create buffers for writing GPU calculations and reading from Javascript
this.#uboBuffer = this.#device.createBuffer({
size: 48,
usage: GPUBufferUsage.UNIFORM | GPUBufferUsage.COPY_DST
})
}
+ static reset (loss?: GPUDeviceLostInfo): void {
+ console.dir(loss)
+ console.warn(`Device lost. Reinitializing...`)
+ this.#cpuBuffer?.destroy()
+ this.#gpuBuffer?.destroy()
+ this.#uboBuffer?.destroy()
+ this.#busy = false
+ this.init()
+ }
+
/**
* Finds a nonce that satisfies the Nano proof-of-work requirements.
*
}
this.#busy = true
if (!/^[A-Fa-f0-9]{64}$/.test(hash)) throw new TypeError(`Invalid hash ${hash}`)
- if (typeof threshold !== 'number') throw new TypeError(`Invalid threshold ${threshold}`)
+ if (typeof threshold !== 'number') throw new TypeError(`Invalid threshold ${threshold}`)
// Ensure WebGPU is initialized before calculating
let loads = 0
if (this.#device == null) throw new Error(`WebGPU device failed to load.`)
let nonce = 0n
- let found = false
- try {
- do {
- // 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))
- }
- const random = Math.floor((Math.random() * 0xffffffff))
- uboView.setUint32(32, random, true)
- uboView.setUint32(36, threshold, true)
- this.#device.queue.writeBuffer(this.#uboBuffer, 0, uboView)
-
- // Reset `found` flag to 0u in WORK before each calculation
- this.#device.queue.writeBuffer(this.#gpuBuffer, 8, new Uint32Array([0]))
-
- // Bind UBO read and GPU write buffers
- const bindGroup = this.#device.createBindGroup({
- layout: this.#bindGroupLayout,
- entries: [
- {
- binding: 0,
- resource: {
- buffer: this.#uboBuffer
- },
+ do {
+ // 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))
+ }
+ const random = Math.floor((Math.random() * 0xffffffff))
+ uboView.setUint32(32, random, true)
+ uboView.setUint32(36, threshold, true)
+ this.#device.queue.writeBuffer(this.#uboBuffer, 0, uboView)
+
+ // Reset `found` flag to 0u in WORK before each calculation
+ this.#device.queue.writeBuffer(this.#gpuBuffer, 8, new Uint32Array([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
- },
+ },
+ {
+ 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(256, 256, 256)
+ 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()])
- // 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(256, 256, 256)
- 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
+ // 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()
- const dataBuffer = this.#cpuBuffer.getMappedRange().slice(0)
+ data = new DataView(this.#cpuBuffer.getMappedRange().slice(0))
+ } catch (err) {
+ console.warn(`Error getting data from GPU. ${err}`)
+ this.reset()
+ } finally {
this.#cpuBuffer.unmap()
+ }
- if (dataBuffer == null) throw new Error(`Failed to get data from buffer.`)
- const dataView = new DataView(dataBuffer)
- nonce = dataView.getBigUint64(0, true)
- found = !!dataView.getUint32(8)
- } while (!found)
- } catch (err) {
- console.warn(`Error getting data from GPU. ${err}`)
- } finally {
- this.#busy = false
- return nonce.toString(16).padStart(16, '0')
- }
+ if (data == null) throw new Error(`Failed to get data from buffer.`)
+ nonce = data.getBigUint64(0, true)
+ this.#busy = !data.getUint32(8)
+ } while (this.#busy)
+ return nonce.toString(16).padStart(16, '0')
}
}
projects / libnemo.git / commitdiff