from timeit import timeit
def runparallel(count=100, num=[]):
numdevices = num
if len(num) == 0: numdevices = range(len(devices))
for i in range(count):
# *****************************************************************
# Load the Tensor to each of the devices
# *****************************************************************
for devnum in numdevices:
img = choice(imgarr)
graphHandle[devnum].LoadTensor(img, 'user object')
# *****************************************************************
# Read the result from each of the devices
# *****************************************************************
for devnum in numdevices:
tensor, userobj = graphHandle[devnum].GetResult()
def runthreaded(count=100,num=[]):
numdevices = num
if len(num) == 0: numdevices = range(len(devices))
thread_list = []
for ii in numdevices:
thread_list.append(Thread(target=runparallel, args=(count,[ii],)))
for thread in thread_list:
thread.start()
for thread in thread_list:
thread.join()
if __name__ == '__main__':
# *****************************************************************
# Runs and times runthreaded with 'i' sticks, until all sticks
# run at once
# *****************************************************************
for i in range(1, len(devices)+1):
num = str(list(range(i)))
tot_time = timeit("runthreaded(count=100,num="+num+")", setup="from __main__ import runthreaded", number=1)
print("\n\nRunning " + argv[1] +" on "+str(i)+" sticks threaded : %0.2f FPS\n\n"%(100.0*i/tot_time))
