index 0b44c7b0cd90707696b20c0b981ea414c395926c..81c9e220f1d71f41c0f1236b75dcc901269a6de5 100755 (executable)
configuration.param_heap_size = (3 << 20)
configuration.network_heap_size = (20 << 20)
-
try:
print('TIDL API: performing one time initialization ...')
- eve = Executor(DeviceType.EVE, eve_device_ids, configuration, 1)
- dsp = Executor(DeviceType.DSP, dsp_device_ids, configuration, 1)
-
# Collect all EOs from EVE and DSP executors
eos = []
- for i in range(eve.get_num_execution_objects()):
- eos.append(eve.at(i))
- for i in range(dsp.get_num_execution_objects()):
- eos.append(dsp.at(i))
+ if len(eve_device_ids) != 0:
+ eve = Executor(DeviceType.EVE, eve_device_ids, configuration, 1)
+ for i in range(eve.get_num_execution_objects()):
+ eos.append(eve.at(i))
+
+ if len(dsp_device_ids) != 0:
+ dsp = Executor(DeviceType.DSP, dsp_device_ids, configuration, 1)
+ for i in range(dsp.get_num_execution_objects()):
+ eos.append(dsp.at(i))
allocate_memory(eos)