Stress test - fix test allocation memory.

* ZE_MEMORY_TYPE_DEVICE requires more memory consumption
oneapi-src · Feb 2, 2024 · 8ff2474 · 8ff2474
1 parent 637ab5a
commit 8ff2474
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/stress_tests/test_memory_allocation/src/test_memory_allocation.cpp b/stress_tests/test_memory_allocation/src/test_memory_allocation.cpp
@@ -143,7 +143,8 @@ TEST_P(
   std::vector<ze_device_memory_properties_t> device_memory_properties =
       lzt::get_memory_properties(device);
 
-  const uint32_t used_vectors_in_test = 3;
+  const uint32_t used_vectors_in_test =
+      test_arguments.memory_type == ZE_MEMORY_TYPE_DEVICE ? 4 : 3;
   uint32_t number_of_dispatches = test_arguments.multiplier;
   uint64_t number_of_all_allocations =
       used_vectors_in_test * number_of_dispatches;