From 5562b47f0687ae019c9ae66bf4692e2a44b0e69d Mon Sep 17 00:00:00 2001
From: Jeff Daily <jeff.daily@amd.com>
Date: Fri, 10 Jun 2022 10:45:28 -0700
Subject: [PATCH] missing #include <thrust/count.h> in
 non_max_suppression_impl.cu (#11730)

Otherwise, depending on cuda or hip thrust versions, transitive header inclusions miss thrust::count_if.
---
 .../providers/cuda/object_detection/non_max_suppression_impl.cu  | 1 +
 1 file changed, 1 insertion(+)

diff --git a/onnxruntime/core/providers/cuda/object_detection/non_max_suppression_impl.cu b/onnxruntime/core/providers/cuda/object_detection/non_max_suppression_impl.cu
index fbb0e9fbb7..79d38319e2 100644
--- a/onnxruntime/core/providers/cuda/object_detection/non_max_suppression_impl.cu
+++ b/onnxruntime/core/providers/cuda/object_detection/non_max_suppression_impl.cu
@@ -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>