saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

missing #include <thrust/count.h> in non_max_suppression_impl.cu (#11730)

Browse source 
Otherwise, depending on cuda or hip thrust versions, transitive header inclusions miss thrust::count_if.
This commit is contained in:
Jeff Daily 2022-06-10 10:45:28 -07:00 committed by GitHub
parent d4ea59654c
commit 5562b47f06
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/core/providers/cuda/object_detection/non_max_suppression_impl.cu
View file

@ -14,6 +14,7 @@ limitations under the License.
==============================================================================*/
/* Modifications Copyright (c) Microsoft. */
#include <thrust/count.h>
#include <thrust/device_vector.h>
#include <thrust/execution_policy.h>