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

Commit 9ec947c

Browse files
gevtushenkogevtushenko
committed
Add block and device merge sort
1 parent ad5299d commit 9ec947c

12 files changed

Lines changed: 3569 additions & 0 deletions

‎cub/agent/agent_merge_sort.cuh‎

Lines changed: 747 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"
@@ -1223,6 +1224,17 @@ public:
12231224

12241225
};
12251226

1227+
template <class Policy,
1228+
class It,
1229+
class T = typename std::iterator_traits<It>::value_type>
1230+
struct BlockLoadType
1231+
{
1232+
using type = cub::BlockLoad<T,
1233+
Policy::BLOCK_THREADS,
1234+
Policy::ITEMS_PER_THREAD,
1235+
Policy::LOAD_ALGORITHM>;
1236+
};
1237+
12261238

12271239
} // CUB namespace
12281240
CUB_NS_POSTFIX // Optional outer namespace(s)

0 commit comments

Comments
 (0)