From 6325970ef8c0ab3abf9c873bc65a1ea42c78e215 Mon Sep 17 00:00:00 2001
From: Cliff Burdick <30670611+cliffburdick@users.noreply.github.com>
Date: Thu, 30 Jan 2025 15:56:59 -0800
Subject: [PATCH] Update deprecated CUB types (#851)

---
 include/matx/transforms/cub.h | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/include/matx/transforms/cub.h b/include/matx/transforms/cub.h
index f2d698df..c5aa9807 100644
--- a/include/matx/transforms/cub.h
+++ b/include/matx/transforms/cub.h
@@ -993,7 +993,7 @@ inline void ExecSort(OutputTensor &a_out,
       if (a.IsContiguous()) {
         cub::DeviceSelect::If(d_temp,
                               temp_storage_bytes,
-                              cub::CountingInputIterator<index_t>{0},
+                              thrust::counting_iterator<index_t>(0),
                               a_out.Data(),
                               cparams_.num_found.Data(),
                               static_cast<int>(TotalSize(a)),
@@ -1004,7 +1004,7 @@ inline void ExecSort(OutputTensor &a_out,
         tensor_impl_t<typename InputOperator::value_type, InputOperator::Rank(), typename InputOperator::desc_type> base = a;
         cub::DeviceSelect::If(d_temp,
                               temp_storage_bytes,
-                              cub::CountingInputIterator<index_t>{0},
+                              thrust::counting_iterator<index_t>(0),
                               a_out.Data(),
                               cparams_.num_found.Data(),
                               static_cast<int>(TotalSize(a)),
@@ -1016,7 +1016,7 @@ inline void ExecSort(OutputTensor &a_out,
     else {
       cub::DeviceSelect::If(d_temp,
                             temp_storage_bytes,
-                            cub::CountingInputIterator<index_t>{0},
+                            thrust::counting_iterator<index_t>(0),
                             a_out.Data(),
                             cparams_.num_found.Data(),
                             static_cast<int>(TotalSize(a)),