Allow more agents & dynamic setting of agent count

src/utils/graphics/get-workgroup-counts.ts

@@ -0,0 +1,28 @@
+export const getWorkgroupCounts = (
+  device: GPUDevice,
+  invocationCount: number,
+  workgroupSize: number
+): [number, number, number] => {
+  const workgroupCount = Math.ceil(invocationCount / workgroupSize);
+
+  const workgroupCountX = Math.min(
+    device.limits.maxComputeWorkgroupsPerDimension,
+    workgroupCount
+  );
+
+  const workgroupCountY = Math.min(
+    device.limits.maxComputeWorkgroupsPerDimension,
+    Math.ceil(workgroupCount / workgroupCountX)
+  );
+
+  const workgroupCountZ = Math.min(
+    device.limits.maxComputeWorkgroupsPerDimension,
+    Math.ceil(workgroupCount / workgroupCountX / workgroupCountY)
+  );
+
+  if (workgroupCountX * workgroupCountY * workgroupCountZ < workgroupCount) {
+    throw new Error('Cannot have this many invocations');
+  }
+
+  return [workgroupCountX, workgroupCountY, workgroupCountZ];
+};

src/utils/graphics/initialize-gpu.ts

@@ -22,10 +22,7 @@ export const initializeGpu = async (): Promise<GPUDevice> => {
     requiredLimits: {
       maxBufferSize: adapter.limits.maxBufferSize,
       maxStorageBufferBindingSize: adapter.limits.maxStorageBufferBindingSize,
-      maxComputeInvocationsPerWorkgroup: adapter.limits.maxComputeInvocationsPerWorkgroup,
-      maxComputeWorkgroupSizeX: adapter.limits.maxComputeWorkgroupSizeX,
-      maxComputeWorkgroupSizeY: adapter.limits.maxComputeWorkgroupSizeY,
-      maxComputeWorkgroupSizeZ: adapter.limits.maxComputeWorkgroupSizeZ,
+      maxComputeWorkgroupsPerDimension: adapter.limits.maxComputeWorkgroupsPerDimension,
     },
   });