gtests: fill_data: add a proper cast

oneapi-src · Jan 24, 2025 · b8270fd · b8270fd
1 parent 0b75fd2
commit b8270fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/gtests/dnnl_test_common.hpp b/tests/gtests/dnnl_test_common.hpp
@@ -443,7 +443,8 @@ inline void fill_data(memory::data_type dt, const memory &mem, float mean,
         return;
     }
 
-    size_t nelems = mem.get_desc().get_size() / dt_size;
+    memory::dim nelems
+            = static_cast<memory::dim>(mem.get_desc().get_size() / dt_size);
     switch (dt) {
         case memory::data_type::f32:
             fill_data<float>(nelems, mem, mean, deviation, sparsity);