Fix mdr error control issue

CODARcode · Nov 15, 2024 · 523203f · 523203f
1 parent 011fc99
commit 523203f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/include/mgard-x/MDR-X/BitplaneEncoder/GroupedBPEncoderGPU.hpp b/include/mgard-x/MDR-X/BitplaneEncoder/GroupedBPEncoderGPU.hpp
@@ -128,6 +128,11 @@ class GroupedEncoderFunctor : public Functor<DeviceType> {
     if (local_bitplane_idx < num_bitplanes + 1) {
       sm_errors[local_bitplane_idx] = 0;
     }
+
+    if (local_data_idx < num_elems_per_TB) {
+      sm_fix_point[local_data_idx] = 0;
+      sm_shifted[local_data_idx] = 0;
+    }
   }
 
   MGARDX_EXEC void Operation2() {

diff --git a/include/mgard-x/RuntimeX/DeviceAdapters/DeviceAdapterOpenmp.h b/include/mgard-x/RuntimeX/DeviceAdapters/DeviceAdapterOpenmp.h
@@ -1416,7 +1416,7 @@ template <> class DeviceCollective<OPENMP> {
                               bool workspace_allocated, int queue_idx) {
 
     if (workspace_allocated) {
-      *result((IDX)0) = std::accumulate(v((IDX)0), v((IDX)n), 0);
+      *result((IDX)0) = std::accumulate(v((IDX)0), v((IDX)n), (T)0);
     } else {
       workspace.resize({(SIZE)1}, queue_idx);
     }

diff --git a/include/mgard-x/RuntimeX/DeviceAdapters/DeviceAdapterSerial.h b/include/mgard-x/RuntimeX/DeviceAdapters/DeviceAdapterSerial.h
@@ -1343,7 +1343,7 @@ template <> class DeviceCollective<SERIAL> {
                               bool workspace_allocated, int queue_idx) {
 
     if (workspace_allocated) {
-      *result((IDX)0) = std::accumulate(v((IDX)0), v((IDX)n), 0);
+      *result((IDX)0) = std::accumulate(v((IDX)0), v((IDX)n), (T)0);
     } else {
       workspace.resize({(SIZE)1}, queue_idx);
     }