Add block and device merge sort

NVIDIA · Jul 7, 2021 · f6d6db6 · f6d6db6
1 parent 1205f88
commit f6d6db6
Show file tree

Hide file tree

Showing 12 changed files with 3,673 additions and 0 deletions.
diff --git a/cub/agent/agent_merge_sort.cuh b/cub/agent/agent_merge_sort.cuh
diff --git a/cub/block/block_load.cuh b/cub/block/block_load.cuh
@@ -34,6 +34,7 @@
 #pragma once
 
 #include <iterator>
+#include <type_traits>
 
 #include "block_exchange.cuh"
 #include "../iterator/cache_modified_input_iterator.cuh"
@@ -1288,6 +1289,17 @@ public:
 
 };
 
+template <class Policy,
+          class It,
+          class T = typename std::iterator_traits<It>::value_type>
+struct BlockLoadType
+{
+  using type = cub::BlockLoad<T,
+                              Policy::BLOCK_THREADS,
+                              Policy::ITEMS_PER_THREAD,
+                              Policy::LOAD_ALGORITHM>;
+};
+
 
 }               // CUB namespace
 CUB_NS_POSTFIX  // Optional outer namespace(s)