Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-07 02:04:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add utils vector mean function

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-14 21:41:18 +02:00
parent 5c8d3f7e25
commit da8f3167cb
4 changed files with 71 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/utils/vector.cu
View File

@@ -61,14 +61,16 @@ void Utils::sum(const float* d_vec, float* d_sum, const unsigned int length) {
}
}
// __device__ float Utils::mean(float* d_vec, const unsigned int length) {
// float sum = 0;
// for (int i = 0; i < length; ++i) {
// sum += d_vec[i];
// }
void Utils::mean(const float* d_vec, float* d_mean, float *d_length, int length) {
Utils::sum(d_vec, d_mean, length);
// void Utils::var(float* d_vec, float* d_mean, float* d_var, const unsigned int length) {
// // TODO:
const int gridSize = (length + BLOCK_SIZE - 1) / BLOCK_SIZE;
Kernels::vec_scalar_div<<<gridSize, BLOCK_SIZE>>>(
d_mean,
d_mean,
d_length,
length
);
// }
CUDA_CHECK(cudaGetLastError());
}