summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to 'examples/pybind/one_eo_per_frame.py')
-rwxr-xr-xexamples/pybind/one_eo_per_frame.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/examples/pybind/one_eo_per_frame.py b/examples/pybind/one_eo_per_frame.py
index dda9712..81c9e22 100755
--- a/examples/pybind/one_eo_per_frame.py
+++ b/examples/pybind/one_eo_per_frame.py
@@ -54,9 +54,6 @@ def main():
54 54
55 num_dsp = Executor.get_num_devices(DeviceType.DSP) 55 num_dsp = Executor.get_num_devices(DeviceType.DSP)
56 num_eve = Executor.get_num_devices(DeviceType.EVE) 56 num_eve = Executor.get_num_devices(DeviceType.EVE)
57 # AM572x default CMEM size is 160MB, 4 EVEs + 2DSPs won't fit
58 if num_eve > 2:
59 num_eve = 2
60 57
61 if num_dsp == 0 and num_eve == 0: 58 if num_dsp == 0 and num_eve == 0:
62 print('No TIDL API capable devices available') 59 print('No TIDL API capable devices available')
@@ -96,7 +93,7 @@ def run(num_eve, num_dsp, configuration):
96 93
97 # Heap sizes for this network determined using Configuration.showHeapStats 94 # Heap sizes for this network determined using Configuration.showHeapStats
98 configuration.param_heap_size = (3 << 20) 95 configuration.param_heap_size = (3 << 20)
99 configuration.network_heap_size = (34 << 20) 96 configuration.network_heap_size = (20 << 20)
100 97
101 try: 98 try:
102 print('TIDL API: performing one time initialization ...') 99 print('TIDL API: performing one time initialization ...')