[SYCL][Graph] Fixes a unitest bug introduced during the fork update (#…

…320) Co-authored-by: Pablo Reble <pablo.reble@intel.com>
reble · Sep 11, 2023 · 20c6b22 · 20c6b22
1 parent d79b04a
commit 20c6b22
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/sycl/unittests/Extensions/CommandGraph.cpp b/sycl/unittests/Extensions/CommandGraph.cpp
@@ -1953,12 +1953,14 @@ TEST_F(MultiThreadGraphTest, BeginEndRecording) {
 
   for (unsigned i = 0; i < NumThreads; ++i) {
     queue MyQueue{Queue.get_context(), Queue.get_device()};
+    GraphRef.begin_recording(MyQueue);
+    runKernels(MyQueue);
     GraphRef.end_recording(MyQueue);
-
-    auto GraphImpl = sycl::detail::getSyclObjImpl(Graph);
-    auto GraphRefImpl = sycl::detail::getSyclObjImpl(GraphRef);
-    ASSERT_EQ(GraphImpl->hasSimilarStructure(GraphRefImpl), true);
   }
+
+  auto GraphImpl = sycl::detail::getSyclObjImpl(Graph);
+  auto GraphRefImpl = sycl::detail::getSyclObjImpl(GraphRef);
+  ASSERT_EQ(GraphImpl->hasSimilarStructure(GraphRefImpl), true);
 }
 
 TEST_F(MultiThreadGraphTest, ExplicitAddNodes) {