Method acquireBuffer

tfjs-backend-webgpu/src/buffer_manager.ts:29–60

(
      size: number, usage: GPUBufferUsageFlags, mappedAtCreation = false,
      reuse = true)

  constructor(private device: GPUDevice) {}

  acquireBuffer(
      size: number, usage: GPUBufferUsageFlags, mappedAtCreation = false,
      reuse = true) {
    let buffer;
    const key = getBufferKey(size, usage);

    if (reuse) {
      if (!this.freeBuffers.has(key)) {
        this.freeBuffers.set(key, []);
      }

      if (this.freeBuffers.get(key).length > 0) {
        buffer = this.freeBuffers.get(key).pop();
        this.numFreeBuffers--;
      } else {
        buffer = this.device.createBuffer({size, usage, mappedAtCreation});
        this.numBytesAllocated += size;
      }
    } else {
      buffer = this.device.createBuffer({size, usage, mappedAtCreation});
      this.numBytesAllocated += size;
    }

    if (!this.usedBuffers.has(key)) {
      this.usedBuffers.set(key, []);
    }
    this.usedBuffers.get(key).push(buffer);
    this.numUsedBuffers++;
    this.numBytesUsed += size;

    return buffer;
  }

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