Skip to content

Commit

Permalink
missing #include <thrust/count.h> in non_max_suppression_impl.cu (#11730
Browse files Browse the repository at this point in the history
)

Otherwise, depending on cuda or hip thrust versions, transitive header inclusions miss thrust::count_if.
  • Loading branch information
jeffdaily authored Jun 10, 2022
1 parent d4ea596 commit 5562b47
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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>

Expand Down

0 comments on commit 5562b47

Please sign in to comment.