saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-12 17:57:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Do not use ADL to invoke std algos (#14851)

Browse source 
This is a follow up for #14716
This commit is contained in:
Yuriy Chernyshov 2023-03-03 03:38:33 +04:00 committed by GitHub
parent 70a31e047a
commit 0ebe8e34f8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/cpu/math/top_k.cc
View file

@ -130,7 +130,7 @@ static void SelectTopK(const Comparator& comparer,
// find the top k (largest or smallest) elements in the data holder - O(n) average. O(n*n) worst case.
// See https://en.wikipedia.org/wiki/Quickselect
nth_element(data_holder.begin(), data_holder.begin() + (k - 1), data_holder.end(), comparer);
std::nth_element(data_holder.begin(), data_holder.begin() + (k - 1), data_holder.end(), comparer);
// sort the top k elements if needed - O (k log k)
if (sort_top_k) {