Skip to content

b12034

Pre-release
Pre-release

Choose a tag to compare

@github-actions github-actions released this 02 Nov 08:02
1c5ed70

webgpu : fix CI by keeping scratch alloc sizes monotonic in batch length (#31422)

  • webgpu: keep scratch alloc sizes monotonic in the batch length

The flash attention vec path and the MMVQ path append a scratch area
after dst, but get_alloc_size only accounted for it when the current
batch takes that path. The size was therefore not monotonic in the
batch length: a graph reserved at n_ubatch (tiled path, no scratch)
or at one token did not cover batches of 2 to 19 query rows, and
GGML_SCHED_NO_REALLOC builds aborted on an unexpected reallocation.

Pass the row count to both path predicates and size the scratch with
the row count clamped to the longest batch the short path serves.

  • nit: align the mmvq pipeline key with clang-format 22

  • webgpu: address review @yomaytk, name the mat-vec row threshold

Website:

Attestations:

macOS/iOS:

Linux:

Android:

Windows:

openEuler:

  • DISABLED
  • openEuler x86 (310p)
  • openEuler x86 (910b, ACL Graph)
  • openEuler aarch64 (310p)
  • openEuler aarch64 (910b, ACL Graph)

UI: