Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion tests/flow/tests_gears_llapi.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,14 +315,18 @@ 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')
self.env.assertEqual(ret[0], b'test2_OK')
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}')
Expand Down
6 changes: 5 additions & 1 deletion tests/flow/tests_llapi.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down