Fix unittest failure

yeonbok · Feb 10, 2025 · 1f95911 · 1f95911
1 parent 815006c
commit 1f95911
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/plugins/intel_gpu/tests/unit/test_cases/reshape_gpu_test.cpp b/src/plugins/intel_gpu/tests/unit/test_cases/reshape_gpu_test.cpp
@@ -97,7 +97,9 @@ void generic_reshape_test(format fmt, tensor const& input_size, tensor const& re
         reshape_size.add(tensor(default_fmt, lower_sizes, 0)).add(tensor(default_fmt, upper_sizes, 0)));
 
     {
-        cldnn::mem_lock<const ElemType> output_ptr(output, get_test_stream());
+        auto output_lockable = engine.allocate_memory(output->get_layout());
+        output->copy_to(get_test_stream(), *output_lockable, true);
+        cldnn::mem_lock<const ElemType> output_ptr(output_lockable, get_test_stream());
         auto output_itr = output_ptr.begin();
 
         auto sizes = reshape_size.sizes(fmt);