fix deprecated

NVIDIA · Jan 21, 2022 · d160454 · d160454
1 parent d4c948b
commit d160454
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cub/agent/agent_unique_by_key.cuh b/cub/agent/agent_unique_by_key.cuh
@@ -33,6 +33,7 @@
 #pragma once
 
 #include <iterator>
+#include <type_traits>
 
 #include "../thread/thread_operators.cuh"
 #include "../block/block_load.cuh"
@@ -108,12 +109,12 @@ struct AgentUniqueByKey
     };
 
     // Cache-modified Input iterator wrapper type (for applying cache modifier) for keys
-    using WrappedKeyInputIteratorT = typename std::conditional<IsPointer<KeyInputIteratorT>::VALUE,
+    using WrappedKeyInputIteratorT = typename std::conditional<std::is_pointer<KeyInputIteratorT>::value,
             CacheModifiedInputIterator<AgentUniqueByKeyPolicyT::LOAD_MODIFIER, KeyT, OffsetT>,     // Wrap the native input pointer with CacheModifiedValuesInputIterator
             KeyInputIteratorT>::type;                                                              // Directly use the supplied input iterator type
 
     // Cache-modified Input iterator wrapper type (for applying cache modifier) for values
-    using WrappedValueInputIteratorT = typename std::conditional<IsPointer<ValueInputIteratorT>::VALUE,
+    using WrappedValueInputIteratorT = typename std::conditional<std::is_pointer<ValueInputIteratorT>::value,
             CacheModifiedInputIterator<AgentUniqueByKeyPolicyT::LOAD_MODIFIER, ValueT, OffsetT>,   // Wrap the native input pointer with CacheModifiedValuesInputIterator
             ValueInputIteratorT>::type;                                                            // Directly use the supplied input iterator type