Support float64 data type as input of benchmark_app (#16435)

openvinotoolkit · Mar 23, 2023 · a20b363 · a20b363
1 parent a205c67
commit a20b363
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/samples/cpp/benchmark_app/remote_tensors_filling.cpp b/samples/cpp/benchmark_app/remote_tensors_filling.cpp
@@ -40,8 +40,10 @@ void fill_buffer_random(void* inputBuffer,
 void fill_buffer(void* inputBuffer, size_t elementsNum, const ov::element::Type& type) {
     if (type == ov::element::f32) {
         fill_buffer_random<float, float>(inputBuffer, elementsNum);
+    } else if (type == ov::element::f64) {
+        fill_buffer_random<double, double>(inputBuffer, elementsNum);
     } else if (type == ov::element::f16) {
-        fill_buffer_random<short, short>(inputBuffer, elementsNum);
+        fill_buffer_random<ov::float16, float>(inputBuffer, elementsNum);
     } else if (type == ov::element::i32) {
         fill_buffer_random<int32_t, int32_t>(inputBuffer, elementsNum);
     } else if (type == ov::element::i64) {