Skip to content

Commit 9ab3cec

Browse files
committed
Add synchronization to autobatch thread
1 parent 52b7e4a commit 9ab3cec

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

test/tests_pytorch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,7 @@ def test_pytorch_modelrun_autobatch(env):
144144
def run():
145145
con = env.getConnection()
146146
con.execute_command('AI.MODELRUN', 'm', 'INPUTS', 'd', 'e', 'OUTPUTS', 'f')
147+
ensureSlaveSynced(con, env)
147148

148149
t = threading.Thread(target=run)
149150
t.start()

test/tests_tensorflow.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,7 @@ def test_run_tf_model_autobatch(env):
393393
def run():
394394
con = env.getConnection()
395395
con.execute_command('AI.MODELRUN', 'm', 'INPUTS', 'd', 'e', 'OUTPUTS', 'f')
396+
ensureSlaveSynced(con, env)
396397

397398
t = threading.Thread(target=run)
398399
t.start()

0 commit comments

Comments
 (0)