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

Implement vector mean calculation

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-28 22:04:15 +02:00
parent 26c12dafc0
commit 0ab623fa23
4 changed files with 79 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/utils/vector.cu
View File

@@ -59,4 +59,16 @@ void Utils::sum(float* d_vec, float* d_sum, const unsigned int length) {
remaining = blocks_needed;
}
}
void Utils::mean(float* d_vec, float* d_mean, const unsigned int length) {
float sum;
Utils::sum(d_vec, &sum, length);
*d_mean = sum / length;
}
void Utils::var(float* d_vec, float* d_mean, float* d_var, const unsigned int length) {
// TODO:
}