From f7f184f27de2aa348e3ce2eef96687d9a1555b4d Mon Sep 17 00:00:00 2001
From: Gonzalo Brito Gadeschi <gonzalob@nvidia.com>
Date: Tue, 4 Jun 2024 06:00:16 -0700
Subject: [PATCH] Fix tbb use_vector

---
 src/tbb/TBBStream.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/tbb/TBBStream.cpp b/src/tbb/TBBStream.cpp
index 9338ae39..01508022 100644
--- 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>