diff --git a/tests/flow/tests_gears_llapi.py b/tests/flow/tests_gears_llapi.py index 910284f6e..f988449b5 100644 --- a/tests/flow/tests_gears_llapi.py +++ b/tests/flow/tests_gears_llapi.py @@ -315,6 +315,10 @@ def test_modelset_modelget_ops(self): self.env.assertEqual(values, [b'2', b'3', b'2', b'3']) def test_modelexecute_op(self): + executions_num = 500 + + if VALGRIND: + executions_num = 10 def multiple_executions(con): ret = con.execute_command('rg.trigger', 'DAGRun_test2') @@ -322,7 +326,7 @@ def multiple_executions(con): values = con.execute_command('AI.TENSORGET', 'test2_res{1}', 'VALUES') self.env.assertEqual(values, [b'4', b'9', b'4', b'9']) - run_test_multiproc(self.env, '{1}', 500, multiple_executions) + run_test_multiproc(self.env, '{1}', executions_num, multiple_executions) def test_scriptexecute_op(self): con = get_connection(self.env, '{1}') diff --git a/tests/flow/tests_llapi.py b/tests/flow/tests_llapi.py index 4f95ca57c..ae2878fc1 100644 --- a/tests/flow/tests_llapi.py +++ b/tests/flow/tests_llapi.py @@ -102,10 +102,14 @@ def test_dag_build_and_run(env): env.assertEqual(ret, b'DAG run success') # Run the DAG LLAPI test again with multi process test to ensure that there are no dead-locks + executions_num = 500 + if VALGRIND: + executions_num = 10 + def run_dag_llapi(con): con.execute_command("RAI_llapi.DAGrun") - run_test_multiproc(env, '{1}', 500, run_dag_llapi) + run_test_multiproc(env, '{1}', executions_num, run_dag_llapi) @with_test_module