Method releaseBuffer

tfjs-backend-webgpu/src/buffer_manager.ts:62–88

(buffer: GPUBuffer, reuse = true)

  }

  releaseBuffer(buffer: GPUBuffer, reuse = true) {
    if (this.freeBuffers.size === 0) {
      return;
    }

    const size = buffer.size;
    const usage = buffer.usage;

    const key = getBufferKey(size, usage);
    const bufferArray = this.usedBuffers.get(key);
    const index = bufferArray.indexOf(buffer);
    if (index < 0) {
      throw new Error('Cannot find the buffer in buffer manager');
    }
    bufferArray[index] = bufferArray[bufferArray.length - 1];
    bufferArray.pop();
    this.numUsedBuffers--;
    this.numBytesUsed -= size;

    if (reuse) {
      this.freeBuffers.get(key).push(buffer);
      this.numFreeBuffers++;
    } else {
      buffer.destroy();
      this.numBytesAllocated -= size;
    }
  }

  getNumUsedBuffers(): number {
    return this.numUsedBuffers;