Skip to content

b9731

Choose a tag to compare

@github-actions github-actions released this 19 Jun 21:54
4b48a53

server : optimize get_token_probabilities (#24796)

Use std::partial_sort to order only the requested top-n tokens instead
of the full vocabulary

logprobs sort: vocab=128000 n_top=0 iters=100
full    sort:   8555.6 us/op
partial sort:    704.3 us/op

Signed-off-by: Adrien Gallouët angt@huggingface.co

macOS/iOS:

Linux:

Android:

Windows:

openEuler:

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

UI: