Set batch size 1 for binary AF matrix tests (#4604)

Fixes #4603

Author: zoddicus

src/webgpu/shader/execution/expression/binary/af_matrix_addition.spec.ts

@@ -8,7 +8,7 @@ import { Type } from '../../../../util/conversion.js';
 import { onlyConstInputSource, run } from '../expression.js';
 
 import { d } from './af_matrix_addition.cache.js';
-import { abstractFloatBinary } from './binary.js';
+import { abstractFloatBinary, kAbstractFloatMatrixBinaryOpBatchSize } from './binary.js';
 
 export const g = makeTestGroup(AllFeaturesMaxLimitsGPUTest);
 
@@ -36,6 +36,7 @@ Accuracy: Correctly rounded
       [Type.mat(cols, rows, Type.abstractFloat), Type.mat(cols, rows, Type.abstractFloat)],
       Type.mat(cols, rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });

src/webgpu/shader/execution/expression/binary/af_matrix_matrix_multiplication.spec.ts

@@ -8,7 +8,7 @@ import { Type } from '../../../../util/conversion.js';
 import { onlyConstInputSource, run } from '../expression.js';
 
 import { d } from './af_matrix_matrix_multiplication.cache.js';
-import { abstractFloatBinary } from './binary.js';
+import { abstractFloatBinary, kAbstractFloatMatrixBinaryOpBatchSize } from './binary.js';
 
 export const g = makeTestGroup(AllFeaturesMaxLimitsGPUTest);
 
@@ -40,6 +40,7 @@ Accuracy: Correctly rounded
       [Type.mat(x_cols, x_rows, Type.abstractFloat), Type.mat(y_cols, y_rows, Type.abstractFloat)],
       Type.mat(y_cols, x_rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });

src/webgpu/shader/execution/expression/binary/af_matrix_scalar_multiplication.spec.ts

@@ -8,7 +8,7 @@ import { Type } from '../../../../util/conversion.js';
 import { onlyConstInputSource, run } from '../expression.js';
 
 import { d } from './af_matrix_scalar_multiplication.cache.js';
-import { abstractFloatBinary } from './binary.js';
+import { abstractFloatBinary, kAbstractFloatMatrixBinaryOpBatchSize } from './binary.js';
 
 export const g = makeTestGroup(AllFeaturesMaxLimitsGPUTest);
 
@@ -36,7 +36,8 @@ Accuracy: Correctly rounded
       [Type.mat(cols, rows, Type.abstractFloat), Type.abstractFloat],
       Type.mat(cols, rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });
 
@@ -64,6 +65,7 @@ Accuracy: Correctly rounded
       [Type.abstractFloat, Type.mat(cols, rows, Type.abstractFloat)],
       Type.mat(cols, rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });

src/webgpu/shader/execution/expression/binary/af_matrix_subtraction.spec.ts

@@ -8,7 +8,7 @@ import { Type } from '../../../../util/conversion.js';
 import { onlyConstInputSource, run } from '../expression.js';
 
 import { d } from './af_matrix_subtraction.cache.js';
-import { abstractFloatBinary } from './binary.js';
+import { abstractFloatBinary, kAbstractFloatMatrixBinaryOpBatchSize } from './binary.js';
 
 export const g = makeTestGroup(AllFeaturesMaxLimitsGPUTest);
 
@@ -36,6 +36,7 @@ Accuracy: Correctly rounded
       [Type.mat(cols, rows, Type.abstractFloat), Type.mat(cols, rows, Type.abstractFloat)],
       Type.mat(cols, rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });

src/webgpu/shader/execution/expression/binary/af_matrix_vector_multiplication.spec.ts

@@ -8,7 +8,7 @@ import { Type } from '../../../../util/conversion.js';
 import { onlyConstInputSource, run } from '../expression.js';
 
 import { d } from './af_matrix_vector_multiplication.cache.js';
-import { abstractFloatBinary } from './binary.js';
+import { abstractFloatBinary, kAbstractFloatMatrixBinaryOpBatchSize } from './binary.js';
 
 export const g = makeTestGroup(AllFeaturesMaxLimitsGPUTest);
 
@@ -36,7 +36,8 @@ Accuracy: Correctly rounded
       [Type.mat(cols, rows, Type.abstractFloat), Type.vec(cols, Type.abstractFloat)],
       Type.vec(rows, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });
 
@@ -64,6 +65,7 @@ Accuracy: Correctly rounded
       [Type.vec(rows, Type.abstractFloat), Type.mat(cols, rows, Type.abstractFloat)],
       Type.vec(cols, Type.abstractFloat),
       t.params,
-      cases
+      cases,
+      kAbstractFloatMatrixBinaryOpBatchSize
     );
   });

src/webgpu/shader/execution/expression/binary/binary.ts

@@ -25,3 +25,6 @@ export function abstractFloatBinary(op: string): ShaderBuilder {
 export function abstractIntBinary(op: string): ShaderBuilder {
   return abstractIntShaderBuilder(values => `(${values.map(v => `(${v})`).join(op)})`);
 }
+
+// See issue #4603 for why using 1 instead of the default
+export const kAbstractFloatMatrixBinaryOpBatchSize = 1;