Lab vectorization 2

dendibakh · Dec 19, 2024 · b5514b8 · b5514b8
1 parent c86b55b
commit b5514b8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/labs/core_bound/vectorization_2/solution.cpp b/labs/core_bound/vectorization_2/solution.cpp
@@ -2,9 +2,14 @@
 
 uint16_t checksum(const Blob &blob) {
   uint16_t acc = 0;
-  for (auto value : blob) {
+  Carry carry = {};
+  for (int i = 0; i < blob.size(); i++) {
+    uint16_t value = blob[i];
     acc += value;
-    acc += acc < value; // add carry
+    carry[i] = acc < value; // save carry
+  }
+  for (int i = 0; i < blob.size(); i++) {
+    acc += carry[i];
   }
   return acc;
 }
diff --git a/labs/core_bound/vectorization_2/solution.hpp b/labs/core_bound/vectorization_2/solution.hpp
@@ -6,6 +6,7 @@
 constexpr std::size_t N = 64 * 1024;
 
 using Blob = std::array<uint16_t, N>;
+using Carry = std::array<uint16_t, N>;
 
 void init(Blob &blob);
 uint16_t checksum(const Blob &blob);