From ded8730d6e9bf1c5efeb2fec1ef254651ca5b38f Mon Sep 17 00:00:00 2001
From: Dmitri Smirnov <yuslepukhin@users.noreply.github.com>
Date: Tue, 28 Jan 2025 08:58:18 -0800
Subject: [PATCH] Remove thrust::unary_function (#23506)

### Description
<!-- Describe your changes. -->
Remove thrust::unary_function which is deprecated in later versions of
CUDA.

### Motivation and Context
Addresses issue: https://github.com/microsoft/onnxruntime/issues/23499
---
 onnxruntime/core/providers/cuda/tensor/compress_impl.cu | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/onnxruntime/core/providers/cuda/tensor/compress_impl.cu b/onnxruntime/core/providers/cuda/tensor/compress_impl.cu
index b2c7b60866..0c04e027ca 100644
--- a/onnxruntime/core/providers/cuda/tensor/compress_impl.cu
+++ b/onnxruntime/core/providers/cuda/tensor/compress_impl.cu
@@ -13,7 +13,6 @@
 
 #include "core/providers/cuda/tensor/compress_impl.h"
 
-#include <thrust/functional.h>
 #include <thrust/iterator/transform_iterator.h>
 
 namespace onnxruntime {
@@ -23,7 +22,7 @@ namespace cuda {
 // in InclusiveSum(). By default, the accumulator type matches the input, but for int8_t
 // the sum overflows quickly, so we want the source type to match the output (int32_t).
 // see https://github.com/NVIDIA/cub/issues/384
-struct CastToInt32 : public thrust::unary_function<int8_t, int32_t> {
+struct CastToInt32 {
   __host__ __device__ int32_t operator()(int8_t v) const {
     return static_cast<int32_t>(v);
   }