diff options
Diffstat (limited to 'examples/pybind/two_eo_per_frame.py')
-rwxr-xr-x | examples/pybind/two_eo_per_frame.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/examples/pybind/two_eo_per_frame.py b/examples/pybind/two_eo_per_frame.py index 953c691..6090d7b 100755 --- a/examples/pybind/two_eo_per_frame.py +++ b/examples/pybind/two_eo_per_frame.py | |||
@@ -55,10 +55,13 @@ def main(): | |||
55 | 55 | ||
56 | # Heap sizes for this network determined using Configuration.showHeapStats | 56 | # Heap sizes for this network determined using Configuration.showHeapStats |
57 | configuration.param_heap_size = (3 << 20) | 57 | configuration.param_heap_size = (3 << 20) |
58 | configuration.network_heap_size = (20 << 20) | 58 | configuration.network_heap_size = (34 << 20) |
59 | 59 | ||
60 | num_dsp = Executor.get_num_devices(DeviceType.DSP) | 60 | num_dsp = Executor.get_num_devices(DeviceType.DSP) |
61 | num_eve = Executor.get_num_devices(DeviceType.EVE) | 61 | num_eve = Executor.get_num_devices(DeviceType.EVE) |
62 | # AM572x default CMEM size is 160MB, 4 EVEs + 2DSPs won't fit | ||
63 | if num_eve > 2: | ||
64 | num_eve = 2 | ||
62 | 65 | ||
63 | if num_dsp == 0 or num_eve == 0: | 66 | if num_dsp == 0 or num_eve == 0: |
64 | print('This example requires EVEs and DSPs.') | 67 | print('This example requires EVEs and DSPs.') |