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

Commit

Permalink
resolve review
Browse files Browse the repository at this point in the history
  • Loading branch information
zasdfgbnm committed Oct 15, 2021
1 parent 53192e0 commit b84e7f1
Showing 1 changed file with 2 additions and 22 deletions.
24 changes: 2 additions & 22 deletions cub/device/dispatch/dispatch_scan_by_key.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,6 @@

#include "../../agent/agent_scan_by_key.cuh"
#include "../../thread/thread_operators.cuh"
// #include "../../grid/grid_queue.cuh"
#include "../../config.cuh"
#include "../../util_debug.cuh"
#include "../../util_device.cuh"
Expand Down Expand Up @@ -125,27 +124,8 @@ struct DeviceScanByKeyPolicy
static constexpr size_t MaxInputBytes = std::max<size_t>(sizeof(KeyT), sizeof(ValueT));
static constexpr size_t CombinedInputBytes = sizeof(KeyT) + sizeof(ValueT);

// SM300
struct Policy300 : ChainedPolicy<300, Policy300, Policy300>
{
enum
{
NOMINAL_4B_ITEMS_PER_THREAD = 6,
ITEMS_PER_THREAD =
Nominal4BItemsToItemsCombined(NOMINAL_4B_ITEMS_PER_THREAD, CombinedInputBytes),
};

typedef AgentScanByKeyPolicy<
128, ITEMS_PER_THREAD,
BLOCK_LOAD_WARP_TRANSPOSE,
LOAD_DEFAULT,
BLOCK_SCAN_WARP_SCANS,
BLOCK_STORE_WARP_TRANSPOSE>
ScanByKeyPolicyT;
};

// SM350
struct Policy350 : ChainedPolicy<350, Policy350, Policy300>
struct Policy350 : ChainedPolicy<350, Policy350, Policy350>
{
enum
{
Expand Down Expand Up @@ -250,7 +230,7 @@ struct DispatchScanByKey:
KeysInputIteratorT d_keys_in, ///< [in] Iterator to the input sequence of key items
ValuesInputIteratorT d_values_in, ///< [in] Iterator to the input sequence of value items
ValuesOutputIteratorT d_values_out, ///< [out] Iterator to the input sequence of value items
EqualityOp equality_op, ///< [in]Binary equality functor
EqualityOp equality_op, ///< [in] Binary equality functor
ScanOpT scan_op, ///< [in] Binary scan functor
InitValueT init_value, ///< [in] Initial value to seed the exclusive scan
OffsetT num_items, ///< [in] Total number of input items (i.e., the length of \p d_in)
Expand Down

0 comments on commit b84e7f1

Please sign in to comment.