tests: benchdnn: fix accesses for int4 types

oneapi-src · Sep 25, 2024 · a65495c · a65495c
1 parent 1692a13
commit a65495c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/benchdnn/dnnl_memory.cpp b/tests/benchdnn/dnnl_memory.cpp
@@ -218,13 +218,13 @@ float dnn_mem_t::get_elem(int64_t idx, int buffer_index) const {
         case dnnl_s4: {
             dnnl::impl::nibble2_t nibble_pair(
                     reinterpret_cast<uint8_t *>(data)[idx / 2]);
-            elem = dnnl::impl::int4_t(nibble_pair.get(idx));
+            elem = dnnl::impl::int4_t(nibble_pair.get(idx % 2));
             break;
         }
         case dnnl_u4: {
             dnnl::impl::nibble2_t nibble_pair(
                     reinterpret_cast<uint8_t *>(data)[idx / 2]);
-            elem = dnnl::impl::uint4_t(nibble_pair.get(idx));
+            elem = dnnl::impl::uint4_t(nibble_pair.get(idx % 2));
             break;
         }
         case dnnl_f4_e2m1: {