From 34663d344a737f80de031f4aaef1c63c51451bf5 Mon Sep 17 00:00:00 2001
From: Mourad Gouicem <mourad.gouicem@intel.com>
Date: Tue, 14 Oct 2025 05:56:53 -0700
Subject: [PATCH] benchdnn: fix reference check in init_memory

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

diff --git a/tests/benchdnn/dnnl_memory.cpp b/tests/benchdnn/dnnl_memory.cpp
index 90e2e93eb8..32cfe13594 100644
--- a/tests/benchdnn/dnnl_memory.cpp
+++ b/tests/benchdnn/dnnl_memory.cpp
@@ -438,6 +438,7 @@ static int init_memory(
     bool is_sycl = is_sycl_engine(engine);
     bool is_opencl = is_opencl_engine(engine);
 
+    if (ret == nullptr) return FAIL;
     *ret = nullptr;
 
     const int nhandles = query_md_num_handles(md);
@@ -470,7 +471,7 @@ static int init_memory(
     // Register device memory
     for (int i = 0; i < nhandles; i++) {
         size_t sz = dnnl_memory_desc_get_size_v2(md, i);
-        if (ret) {
+        if (*ret) {
             DNN_SAFE(
                     dnnl_memory_get_data_handle_v2(*ret, &handles[i], i), CRIT);
         }