wgsl: Use selectNCases to restrict the number of AF Matrix cases

Fixes #3457
gpuweb · Mar 7, 2024 · 6e6a92f · 6e6a92f
1 parent adaa479
commit 6e6a92f
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 12 deletions.
diff --git a/src/resources/cache/hashes.json b/src/resources/cache/hashes.json
@@ -2,10 +2,10 @@
   "webgpu/shader/execution/binary/af_addition.bin": "1a138e40",
   "webgpu/shader/execution/binary/af_logical.bin": "2a334d4b",
   "webgpu/shader/execution/binary/af_division.bin": "801a63d2",
-  "webgpu/shader/execution/binary/af_matrix_addition.bin": "d392437a",
+  "webgpu/shader/execution/binary/af_matrix_addition.bin": "5c7c1cc9",
   "webgpu/shader/execution/binary/af_matrix_matrix_multiplication.bin": "8a56ca30",
   "webgpu/shader/execution/binary/af_matrix_scalar_multiplication.bin": "3b36edc4",
-  "webgpu/shader/execution/binary/af_matrix_subtraction.bin": "c2a167d1",
+  "webgpu/shader/execution/binary/af_matrix_subtraction.bin": "ab2b8211",
   "webgpu/shader/execution/binary/af_matrix_vector_multiplication.bin": "3bdd724b",
   "webgpu/shader/execution/binary/af_multiplication.bin": "af6472be",
   "webgpu/shader/execution/binary/af_remainder.bin": "582570bc",

diff --git a/src/resources/cache/webgpu/shader/execution/binary/af_matrix_addition.bin b/src/resources/cache/webgpu/shader/execution/binary/af_matrix_addition.bin
diff --git a/src/resources/cache/webgpu/shader/execution/binary/af_matrix_subtraction.bin b/src/resources/cache/webgpu/shader/execution/binary/af_matrix_subtraction.bin
diff --git a/src/webgpu/shader/execution/expression/binary/af_matrix_addition.cache.ts b/src/webgpu/shader/execution/expression/binary/af_matrix_addition.cache.ts
@@ -1,17 +1,22 @@
 import { FP } from '../../../../util/floating_point.js';
 import { sparseMatrixF64Range } from '../../../../util/math.js';
+import { selectNCases } from '../case.js';
 import { makeCaseCache } from '../case_cache.js';
 
 // Cases: matCxR
 const mat_cases = ([2, 3, 4] as const)
   .flatMap(cols =>
     ([2, 3, 4] as const).map(rows => ({
       [`mat${cols}x${rows}`]: () => {
-        return FP.abstract.generateMatrixPairToMatrixCases(
-          sparseMatrixF64Range(cols, rows),
-          sparseMatrixF64Range(cols, rows),
-          'finite',
-          FP.abstract.additionMatrixMatrixInterval
+        return selectNCases(
+          'binary/af_matrix_addition',
+          50,
+          FP.abstract.generateMatrixPairToMatrixCases(
+            sparseMatrixF64Range(cols, rows),
+            sparseMatrixF64Range(cols, rows),
+            'finite',
+            FP.abstract.additionMatrixMatrixInterval
+          )
         );
       },
     }))

diff --git a/src/webgpu/shader/execution/expression/binary/af_matrix_subtraction.cache.ts b/src/webgpu/shader/execution/expression/binary/af_matrix_subtraction.cache.ts
@@ -1,17 +1,22 @@
 import { FP } from '../../../../util/floating_point.js';
 import { sparseMatrixF64Range } from '../../../../util/math.js';
+import { selectNCases } from '../case.js';
 import { makeCaseCache } from '../case_cache.js';
 
 // Cases: matCxR
 const mat_cases = ([2, 3, 4] as const)
   .flatMap(cols =>
     ([2, 3, 4] as const).map(rows => ({
       [`mat${cols}x${rows}`]: () => {
-        return FP.abstract.generateMatrixPairToMatrixCases(
-          sparseMatrixF64Range(cols, rows),
-          sparseMatrixF64Range(cols, rows),
-          'finite',
-          FP.abstract.subtractionMatrixMatrixInterval
+        return selectNCases(
+          'binary/af_matrix_subtraction',
+          50,
+          FP.abstract.generateMatrixPairToMatrixCases(
+            sparseMatrixF64Range(cols, rows),
+            sparseMatrixF64Range(cols, rows),
+            'finite',
+            FP.abstract.subtractionMatrixMatrixInterval
+          )
         );
       },
     }))