diff options
Diffstat (limited to 'examples/pybind/one_eo_per_frame.py')
-rwxr-xr-x | examples/pybind/one_eo_per_frame.py | 5 |
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 ...') |