From a65495ce11df51c90f2b84ffc826129d26a35551 Mon Sep 17 00:00:00 2001
From: "Kassen, Andrew" <andrew.kassen@intel.com>
Date: Mon, 23 Sep 2024 10:59:10 -0700
Subject: [PATCH] tests: benchdnn: fix accesses for int4 types

---
 tests/benchdnn/dnnl_memory.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/benchdnn/dnnl_memory.cpp b/tests/benchdnn/dnnl_memory.cpp
index 12ae03cc710..547b7f3d483 100644
--- 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: {