Allows the specification of workgroups for compute shaders and the specification of the dispatchSize

Author: Attila Schroeder

src/nodes/gpgpu/ComputeNode.js

@@ -219,7 +219,7 @@ export default ComputeNode;
  * @tsl
  * @function
  * @param {Node} node - TODO
- * @param {number} count - TODO.
+ * @param {number} countOrWorkgroupSize - TODO.
  * @param {Array<number>} [workgroupSize=[ 64, 1, 1 ]]
  * @returns {AtomicFunctionNode}
  */

src/renderers/common/Renderer.js

@@ -2288,6 +2288,7 @@ class Renderer {
 	 * if the renderer has been initialized.
 	 *
 	 * @param {Node|Array<Node>} computeNodes - The compute node(s).
+	 * @param {Array<number>} dispatchSize - Array with [x,y,z] values for dispatch.
 	 * @return {Promise|undefined} A Promise that resolve when the compute has finished. Only returned when the renderer has not been initialized.
 	 */
 	compute( computeNodes, dispatchSize = [ 0, 0, 0 ] ) {

src/renderers/webgpu/WebGPUBackend.js

@@ -1318,6 +1318,7 @@ class WebGPUBackend extends Backend {
 	 * @param {Node} computeNode - The compute node.
 	 * @param {Array<BindGroup>} bindings - The bindings.
 	 * @param {ComputePipeline} pipeline - The compute pipeline.
+	 * @param {Array<number>} dispatchSize - Array with [x,y,z] values for dispatch.
 	 */
 	compute( computeGroup, computeNode, bindings, pipeline, dispatchSize ) {
 

src/renderers/webgpu/nodes/WGSLNodeBuilder.js

@@ -1902,7 +1902,7 @@ ${ flowData.code }
 
 			const workgroupSize = this.object.workgroupSize || [ 8, 8, 1 ];
 
-			if ( workgroupSize.length !== 3 ) throw new Error( "workgroupSize must have 3 elements" );
+			if ( workgroupSize.length !== 3 ) throw new Error( 'workgroupSize must have 3 elements' );
 
 			this.computeShader = this._getWGSLComputeCode( shadersData.compute, workgroupSize );