frozenleaves/oneDNN
1
0
Fork 0
mirror of https://github.com/uxlfoundation/oneDNN.git synced 2025-10-20 18:43:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

benchdnn: fix reference check in init_memory

Browse Source
This commit is contained in:
Mourad Gouicem
2025-10-14 05:56:53 -07:00
parent 67cd11b889
commit 34663d344a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/benchdnn/dnnl_memory.cpp
View File

@ -438,6 +438,7 @@ static int init_memory(
bool is_sycl = is_sycl_engine(engine); bool is_sycl = is_sycl_engine(engine);
bool is_opencl = is_opencl_engine(engine); bool is_opencl = is_opencl_engine(engine);
if (ret == nullptr) return FAIL;
*ret = nullptr; *ret = nullptr;
const int nhandles = query_md_num_handles(md); const int nhandles = query_md_num_handles(md);
@ -470,7 +471,7 @@ static int init_memory(
// Register device memory // Register device memory
for (int i = 0; i < nhandles; i++) { for (int i = 0; i < nhandles; i++) {
size_t sz = dnnl_memory_desc_get_size_v2(md, i); size_t sz = dnnl_memory_desc_get_size_v2(md, i);
if (ret) { if (*ret) {
DNN_SAFE( DNN_SAFE(
dnnl_memory_get_data_handle_v2(*ret, &handles[i], i), CRIT); dnnl_memory_get_data_handle_v2(*ret, &handles[i], i), CRIT);
} }