Add block and device merge sort

seunghwak · Jul 28, 2021 · 267d78f · 267d78f
1 parent 94a50bf
commit 267d78f
Show file tree

Hide file tree

Showing 12 changed files with 3,663 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"
@@ -1284,6 +1285,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_END