aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to 'examples/pybind/two_eo_per_frame.py')
-rwxr-xr-xexamples/pybind/two_eo_per_frame.py5
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.')