fix thread safe

Signed-off-by: Mishin, Ilya <ilya.mishin@intel.com>
uxlfoundation · Aug 11, 2021 · 64d8acc · 64d8acc
1 parent 3dfc84e
commit 64d8acc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/conformance/conformance_graph.cpp b/test/conformance/conformance_graph.cpp
@@ -76,7 +76,6 @@ void test_functional_nodes_rf_reset_protocol(){
 
     f.try_put(0);
     f.try_put(0);
-
     CHECK_MESSAGE((counting_body.execute_count == 0), "Body should not be executed");
     g.reset(oneapi::tbb::flow::rf_reset_protocol);
 
@@ -94,7 +93,7 @@ void test_buffering_nodes_rf_reset_protocol(Args... node_body){
     CHECK_MESSAGE((tmp == -1), "Value should not be updated");
 
     testing_node.try_put(1);
-
+    g.wait_for_all();
     g.reset(oneapi::tbb::flow::rf_reset_protocol);
 
     tmp = -1;