alternate 'enabled' mask element value - 1,2

Signed-off-by: Konstantin S Bobrovsky <[email protected]>

Author: kbobrovs

SYCL/ESIMD/api/esimd_merge.cpp

@@ -21,6 +21,14 @@ using namespace sycl::ext::intel::experimental::esimd;
 using namespace sycl::ext::intel::experimental;
 using namespace cl::sycl;
 
+template <class T> void prn(T *arr, int size, const char *title) {
+  std::cout << title << ": ";
+  for (int i = 0; i < size; ++i) {
+    std::cout << " " << arr[i];
+  }
+  std::cout << "\n";
+}
+
 int main(void) {
   constexpr unsigned VL = 16;
   constexpr unsigned FACTOR = 2;
@@ -42,10 +50,11 @@ int main(void) {
   constexpr int VAL2 = 3;
 
   for (int i = 0; i < Size; ++i) {
-    A[i] = i % 2 + VAL1; // 1 2 1 2 ...
-    B[i] = i % 2 + VAL2; // 3 4 3 4 ...
-    // mask out first half of sub-vector:
-    M[i] = (i % SUB_VL) >= (SUB_VL / 2) ? 1 : 0; // 0 0 1 1 0 0 1 1 ...
+    A[i] = i % 2 + VAL1; // 1212 ...
+    B[i] = i % 2 + VAL2; // 3434 ...
+    // mask out first half of sub-vector, alternating '1' and '2' as 'enabled'
+    // bit representation in a mask element:
+    M[i] = (i % SUB_VL) >= (SUB_VL / 2) ? (i % SUB_VL - 1) : 0; // 00120012 ...
     C[i] = VAL0;
   }
 
@@ -57,7 +66,7 @@ int main(void) {
         simd_mask<SUB_VL> m(M + i * VL);
         // va:         1212121212121212
         // vb:         3434343434343434
-        // m:          0011001100110011
+        // m:          0012001200120012
         // va.sel.sel: 1111
         // vb.sel.sel: 4444
         // vc:         1144
@@ -74,15 +83,16 @@ int main(void) {
     sycl::free(A, q);
     sycl::free(B, q);
     sycl::free(C, q);
+    sycl::free(M, q);
     return 1;
   }
 
   int err_cnt = 0;
 
-  for (int i = 0; i < Size; ++i) {
-    std::cout << " " << C[i];
-  }
-  std::cout << "\n";
+  prn(A, Size, "A");
+  prn(B, Size, "B");
+  prn(M, Size, "M");
+  prn(C, Size, "C");
 
   for (int i = 0; i < Size; ++i) {
     int j = i % VL;
@@ -105,6 +115,7 @@ int main(void) {
   sycl::free(A, q);
   sycl::free(B, q);
   sycl::free(C, q);
+  sycl::free(M, q);
   std::cout << (err_cnt > 0 ? "FAILED\n" : "Passed\n");
   return err_cnt > 0 ? 1 : 0;
 }