Skip to content
This repository was archived by the owner on Mar 21, 2024. It is now read-only.

Commit 9631ad6

Browse files
gevtushenkogevtushenko
committed
Add block and device merge sort
1 parent e539681 commit 9631ad6

12 files changed

Lines changed: 3663 additions & 0 deletions

‎cub/agent/agent_merge_sort.cuh‎

Lines changed: 750 additions & 0 deletions
Large diffs are not rendered by default.

‎cub/block/block_load.cuh‎

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
#pragma once
3535

3636
#include <iterator>
37+
#include <type_traits>
3738

3839
#include "block_exchange.cuh"
3940
#include "../iterator/cache_modified_input_iterator.cuh"
@@ -1284,6 +1285,17 @@ public:
12841285

12851286
};
12861287

1288+
template <class Policy,
1289+
class It,
1290+
class T = typename std::iterator_traits<It>::value_type>
1291+
struct BlockLoadType
1292+
{
1293+
using type = cub::BlockLoad<T,
1294+
Policy::BLOCK_THREADS,
1295+
Policy::ITEMS_PER_THREAD,
1296+
Policy::LOAD_ALGORITHM>;
1297+
};
1298+
12871299

12881300
CUB_NAMESPACE_END
12891301

0 commit comments

Comments
 (0)