wgblas 0.1.2 → 1.0.0

package/src/util/workgroup.mjs

@@ -9,23 +9,23 @@ const WORKGROUP_SIZE_2D = 8;
  * Calculates the number of workgroups to dispatch, clamped to the device's
  * `maxComputeWorkgroupsPerDimension` limit (default 65535 across most devices).
  *
- * - 1D (pass only `dim1`): returns a single count for `dispatchWorkgroups(n)`.
- * - 2D (pass both `dim1` and `dim2`): returns `{ x, y }` for `dispatchWorkgroups(x, y)`.
- *   `dim1` maps to rows (y) and `dim2` maps to columns (x).
+ * - 1D (pass only `rows`): returns a single count for `dispatchWorkgroups(n)`.
+ * - 2D (pass both `rows` and `cols`): returns `{ x, y }` for `dispatchWorkgroups(x, y)`.
+ *   `rows` maps to the y dimension and `cols` maps to the x dimension.
  *
- * @param {number} dim1 - row count (1D: element count)
- * @param {number} [dim2] - column count; omit for a 1D dispatch
+ * @param {number} rows - row count (1D: element count)
+ * @param {number} [cols] - column count; omit for a 1D dispatch
  * @returns {number | { x: number, y: number }}
  * @see {@link https://developer.mozilla.org/en-US/docs/Web/API/GPUSupportedLimits GPUSupportedLimits} (`maxComputeWorkgroupsPerDimension`)
  */
-export function calcWorkgroups(dim1, dim2) {
+export function calcWorkgroups(rows, cols) {
   const max = getDevice().limits.maxComputeWorkgroupsPerDimension;
-  if (dim2 === undefined) {
-    return Math.min(Math.ceil(dim1 / WORKGROUP_SIZE_1D), max);
+  if (cols === undefined) {
+    return Math.min(Math.ceil(rows / WORKGROUP_SIZE_1D), max);
   } else {
     return {
-      x: Math.min(Math.ceil(dim2 / WORKGROUP_SIZE_2D), max),
-      y: Math.min(Math.ceil(dim1 / WORKGROUP_SIZE_2D), max),
+      x: Math.min(Math.ceil(cols / WORKGROUP_SIZE_2D), max),
+      y: Math.min(Math.ceil(rows / WORKGROUP_SIZE_2D), max),
     };
   }
 }