Fix tbb use_vector

gonzalobg · Jun 4, 2024 · f7f184f · f7f184f
1 parent d787351
commit f7f184f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/tbb/TBBStream.cpp b/src/tbb/TBBStream.cpp
@@ -59,9 +59,15 @@ void TBBStream<T>::init_arrays(T initA, T initB, T initC)
 template <class T>
 void TBBStream<T>::get_arrays(T const*& h_a, T const*& h_b, T const*& h_c)
 {
+#ifdef USE_VECTOR
+  h_a = a.data();
+  h_b = b.data();
+  h_c = c.data();
+#else
   h_a = a;
   h_b = b;
   h_c = c;
+#endif
 }
 
 template <class T>