,

src/pipelines/agents/agent-generation/agent-generation-pipeline.ts

@@ -26,6 +26,7 @@ export class AgentGenerationPipeline {
   private readonly countingPipeline: GPUComputePipeline;
   private readonly resizePipeline: GPUComputePipeline;
   private readonly compactionPipeline: GPUComputePipeline;
+  private readonly compactedAgentsCopyPipeline: GPUComputePipeline;
 
   public readonly agentsBuffer: GPUBuffer;
   private readonly compactedAgentsBuffer: GPUBuffer;
@@ -109,7 +110,7 @@ export class AgentGenerationPipeline {
 
     this.compactedAgentsBuffer = this.device.createBuffer({
       size: this.maxAgentCount * AGENT_SIZE_IN_BYTES,
-      usage: GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC,
+      usage: GPUBufferUsage.STORAGE,
     });
 
     this.countersBuffer = this.device.createBuffer({
@@ -216,20 +217,32 @@ export class AgentGenerationPipeline {
       },
     });
 
+    const compactionModule = smartCompile(
+      device,
+      CommonState.shaderCode,
+      agentSchema,
+      compactionShader
+    );
+
     this.compactionPipeline = device.createComputePipeline({
       layout: device.createPipelineLayout({
         bindGroupLayouts: [commonState.bindGroupLayout, this.compactionBindGroupLayout],
       }),
       compute: {
-        module: smartCompile(
-          device,
-          CommonState.shaderCode,
-          agentSchema,
-          compactionShader
-        ),
+        module: compactionModule,
         entryPoint: 'main',
       },
     });
+
+    this.compactedAgentsCopyPipeline = device.createComputePipeline({
+      layout: device.createPipelineLayout({
+        bindGroupLayouts: [commonState.bindGroupLayout, this.compactionBindGroupLayout],
+      }),
+      compute: {
+        module: compactionModule,
+        entryPoint: 'copyCompactedAgents',
+      },
+    });
   }
 
   public get maxAgentCount(): number {
@@ -364,13 +377,19 @@ export class AgentGenerationPipeline {
     );
     passEncoder.end();
 
-    commandEncoder.copyBufferToBuffer(
-      this.compactedAgentsBuffer,
-      0,
-      this.agentsBuffer,
-      0,
-      agentCount * AGENT_SIZE_IN_BYTES
+    const copyPassEncoder = commandEncoder.beginComputePass();
+    copyPassEncoder.setPipeline(this.compactedAgentsCopyPipeline);
+    this.commonState.execute(copyPassEncoder);
+    copyPassEncoder.setBindGroup(1, this.compactionBindGroup);
+    copyPassEncoder.dispatchWorkgroups(
+      ...getWorkgroupCounts(
+        this.device,
+        agentCount,
+        AgentGenerationPipeline.WORKGROUP_SIZE
+      )
     );
+    copyPassEncoder.end();
+
     commandEncoder.copyBufferToBuffer(
       this.countersBuffer,
       0,