From cdfb14b83e4c74cbc40a185bff3535feca738e5f Mon Sep 17 00:00:00 2001
From: Tobias Winchen <tobias.winchen@rwth-aachen.de>
Date: Fri, 27 Aug 2021 11:01:32 +0000
Subject: [PATCH] Fix baseline NaN for no heaps in gate
---
psrdada_cpp/effelsberg/edd/src/GatedSpectrometer.cu | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/psrdada_cpp/effelsberg/edd/src/GatedSpectrometer.cu b/psrdada_cpp/effelsberg/edd/src/GatedSpectrometer.cu
index 8c1c7ec9..663654d2 100644
--- a/psrdada_cpp/effelsberg/edd/src/GatedSpectrometer.cu
+++ b/psrdada_cpp/effelsberg/edd/src/GatedSpectrometer.cu
@@ -116,8 +116,12 @@ __global__ void update_baselines(float* __restrict__ baseLineG0,
NG1 += stats_G1[i];
}
- baseLineG0[0] = baseLineNG0[0] / NG0;
- baseLineG1[0] = baseLineNG1[0] / NG1;
+ if (NG0 > 0)
+ baseLineG0[0] = baseLineNG0[0] / NG0;
+
+ if (NG1 > 0)
+ baseLineG1[0] = baseLineNG1[0] / NG1;
+
baseLineNG0[0] = 0;
baseLineNG1[0] = 0;
}
--
GitLab