Merge pull request #322 from senior-zero/main-feature/github/cub_merg…

…e_sort Add CUB block and device merge sort
NVIDIA · Jul 28, 2021 · e59cff3 · e59cff3
2 parents 94a50bf + 267d78f
commit e59cff3
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