@@ -6,14 +6,10 @@
|
|||||||
// CHECK: #include <hipcub/hipcub.hpp>
|
// CHECK: #include <hipcub/hipcub.hpp>
|
||||||
#include <cub/cub.cuh>
|
#include <cub/cub.cuh>
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
|
|
||||||
// using namespace hipcub;
|
// using namespace hipcub;
|
||||||
using namespace cub;
|
using namespace cub;
|
||||||
|
|
||||||
/**
|
// Simple CUDA kernel for computing tiled partial sums
|
||||||
* Simple CUDA kernel for computing tiled partial sums
|
|
||||||
*/
|
|
||||||
template <int BLOCK_THREADS, int ITEMS_PER_THREAD,
|
template <int BLOCK_THREADS, int ITEMS_PER_THREAD,
|
||||||
BlockLoadAlgorithm LOAD_ALGO,
|
BlockLoadAlgorithm LOAD_ALGO,
|
||||||
BlockScanAlgorithm SCAN_ALGO>
|
BlockScanAlgorithm SCAN_ALGO>
|
||||||
|
|||||||
Reference in New Issue
Block a user