[INFRA] Import NVIDIA/CCCL upstream as optimization reference library
CCCL (CUDA C++ Core Libraries) provides: - CUB: device/block/warp-level GPU primitives (reduce, scan, sort, topk) - Thrust: high-level parallel algorithms (transform_reduce, sort, scan) - libcudacxx: CUDA C++ standard library (atomics, barriers, memory) - cudax: experimental features (memory resources, allocators) - Tuning policies: per-SM hardware-specific algorithm parameters Competition optimization vectors mapped to CCCL: - Output TPS (83% weight): warp_reduce, block_reduce, device_topk - Input TPS (14% weight): device_scan, block_load, prefetch - Cache TPS (3% weight): prefix caching strategy patterns - Memory (0.9 util): pooled/cached/buddy allocators Source: https://github.com/NVIDIA/cccl (shallow clone, HEAD only) License: Apache-2.0
This commit is contained in:
101
cccl_upstream/thrust/examples/simple_moving_average.cu
Normal file
101
cccl_upstream/thrust/examples/simple_moving_average.cu
Normal file
@@ -0,0 +1,101 @@
|
||||
#include <thrust/device_vector.h>
|
||||
#include <thrust/functional.h>
|
||||
#include <thrust/host_vector.h>
|
||||
#include <thrust/random.h>
|
||||
#include <thrust/scan.h>
|
||||
#include <thrust/sequence.h>
|
||||
#include <thrust/transform.h>
|
||||
|
||||
#include <iomanip>
|
||||
#include <iostream>
|
||||
|
||||
// Efficiently computes the simple moving average (SMA) [1] of a data series
|
||||
// using a parallel prefix-sum or "scan" operation.
|
||||
//
|
||||
// Note: additional numerical precision should be used in the cumulative summing
|
||||
// stage when computing the SMA of large data series. The most straightforward
|
||||
// remedy is to replace 'float' with 'double'. Alternatively a Kahan or
|
||||
// "compensated" summation algorithm could be applied [2].
|
||||
//
|
||||
// [1] http://en.wikipedia.org/wiki/Moving_average#Simple_moving_average
|
||||
// [2] http://en.wikipedia.org/wiki/Kahan_summation_algorithm
|
||||
|
||||
// compute the difference of two positions in the cumumulative sum and
|
||||
// divide by the SMA window size w.
|
||||
template <typename T>
|
||||
struct minus_and_divide
|
||||
{
|
||||
T w;
|
||||
|
||||
minus_and_divide(T w)
|
||||
: w(w)
|
||||
{}
|
||||
|
||||
__host__ __device__ T operator()(const T& a, const T& b) const
|
||||
{
|
||||
return (a - b) / w;
|
||||
}
|
||||
};
|
||||
|
||||
template <typename InputVector, typename OutputVector>
|
||||
void simple_moving_average(const InputVector& data, size_t w, OutputVector& output)
|
||||
{
|
||||
using T = typename InputVector::value_type;
|
||||
|
||||
if (data.size() < w)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
// allocate storage for cumulative sum
|
||||
thrust::device_vector<T> temp(data.size() + 1);
|
||||
|
||||
// compute cumulative sum
|
||||
thrust::exclusive_scan(data.begin(), data.end(), temp.begin());
|
||||
temp[data.size()] = data.back() + temp[data.size() - 1];
|
||||
|
||||
// compute moving averages from cumulative sum
|
||||
thrust::transform(temp.begin() + w, temp.end(), temp.begin(), output.begin(), minus_and_divide<T>(T(w)));
|
||||
}
|
||||
|
||||
int main()
|
||||
{
|
||||
// length of data series
|
||||
size_t n = 30;
|
||||
|
||||
// window size of the moving average
|
||||
size_t w = 4;
|
||||
|
||||
// generate random data series
|
||||
thrust::host_vector<float> host_data(n);
|
||||
thrust::default_random_engine rng;
|
||||
thrust::uniform_int_distribution<int> dist(0, 10);
|
||||
for (auto& e : host_data)
|
||||
{
|
||||
e = static_cast<float>(dist(rng));
|
||||
}
|
||||
thrust::device_vector<float> data = host_data;
|
||||
|
||||
// allocate storage for averages
|
||||
thrust::device_vector<float> averages(data.size() - (w - 1));
|
||||
|
||||
// compute SMA using standard summation
|
||||
simple_moving_average(data, w, averages);
|
||||
|
||||
// print data series
|
||||
std::cout << "data series: [ ";
|
||||
for (const auto& value : data)
|
||||
{
|
||||
std::cout << value << " ";
|
||||
}
|
||||
std::cout << "]" << '\n';
|
||||
|
||||
// print moving averages
|
||||
std::cout << "simple moving averages (window = " << w << ")" << '\n';
|
||||
for (size_t i = 0; i < averages.size(); i++)
|
||||
{
|
||||
std::cout << " [" << std::setw(2) << i << "," << std::setw(2) << (i + w) << ") = " << averages[i] << '\n';
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user