djairoh/cuda-based-raytrace
1
0
Fork 0
mirror of https://github.com/MartinOpat/cuda-based-raytrace.git synced 2025-06-07 02:13:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

FieldData structs implemented

Browse Source
This commit is contained in:
Robin
2024-12-20 18:46:18 +01:00
parent ac63abe93a
commit 0db764c52f
10 changed files with 196 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/main.cu
View File

@@ -1,19 +1,31 @@
#include "hurricanedata/datareader.h"
#include "hurricanedata/gpubuffer.h"
#include <cuda_runtime.h>
#include <device_launch_parameters.h>
#include <iostream>
#include <cmath>
__device__ float getVal(
const FieldMetadata &md,
const FieldData &d,
const size_t &timeInd,
const size_t &lonInd,
const size_t &latInd,
const size_t &levInd
) {
return d.valArrays[0][timeInd];
}
// Not parallel computation
__global__ void computeMean(float *ans, size_t *masked_vals, size_t n, float *x) {
__global__ void computeMean(float *ans, size_t *masked_vals, const FieldMetadata &fmd, FieldData fd) {
float sum = 0;
size_t num_not_masked_values = 0;
size_t num_masked_values = 0;
for (int i = 0; i < n; i++) {
if (x[i] < 1E14) { /* If x is not missing value */
for (int i = 0; i < fmd.widthSize*fmd.heightSize*fmd.depthSize*fd.timeSize; i++) {
double xi = getVal(fmd, fd, i, 0, 0, 0);
if (xi < 1E14) { /* If x is not missing value */
num_not_masked_values++;
sum += x[i];
sum += xi;
} else {
num_masked_values++;
}
@@ -25,7 +37,9 @@ __global__ void computeMean(float *ans, size_t *masked_vals, size_t n, float *x)
int main() {
std::string path = "data/MERRA2_400.inst6_3d_ana_Np.20120101.nc4";
std::string variable = "T";
auto arr = loadDataToDevice(path, variable);
GPUBuffer buffer{path, variable};
auto fd = buffer.nextFieldData();
float *ptr_mean;
cudaMallocManaged(&ptr_mean, sizeof(float));
@@ -33,15 +47,14 @@ int main() {
size_t *ptr_masked;
cudaMallocManaged(&ptr_masked, sizeof(size_t));
computeMean<<<1, 1>>>(ptr_mean, ptr_masked, arr.second, arr.first);
computeMean<<<1, 1>>>(ptr_mean, ptr_masked, *buffer.fmd, fd);
cudaDeviceSynchronize();
std::cout << "Mean = " << *ptr_mean << " calculated from " << arr.second << " values where " << *ptr_masked << " are masked values.\n";
std::cout << "Mean = " << *ptr_mean << " values where " << *ptr_masked << " are masked values.\n";
cudaFree(arr.first);
cudaFree(fd.valArrays[0]);
cudaFree(ptr_mean);
cudaFree(ptr_masked);
return 0;
}