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, 1 insertions, 4 deletions
diff --git a/examples/pybind/two_eo_per_frame.py b/examples/pybind/two_eo_per_frame.py index 6090d7b..953c691 100755 --- a/examples/pybind/two_eo_per_frame.py +++ b/examples/pybind/two_eo_per_frame.py | |||
@@ -55,13 +55,10 @@ 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 = (34 << 20) | 58 | configuration.network_heap_size = (20 << 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 | ||
65 | 62 | ||
66 | if num_dsp == 0 or num_eve == 0: | 63 | if num_dsp == 0 or num_eve == 0: |
67 | print('This example requires EVEs and DSPs.') | 64 | print('This example requires EVEs and DSPs.') |