Skip to content

Commit b85e1fb

Browse files
Fix EDP optimization returning all energy-or-latency-optimal mappings.
1 parent 336b4ff commit b85e1fb

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

accelforge/mapper/FFM/_join_pmappings/join_pmappings.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -349,6 +349,9 @@ def clean_compress_and_join_pmappings(
349349
joined = decompress_pmappings(joined, decompress_data)
350350

351351
_apply_edp_columns(joined.data, metrics)
352+
# Pareto prune again in case the EDP column application reduced number of
353+
# objectives.
354+
joined.make_pareto()
352355

353356
for einsum_name in einsum2pmappings:
354357
col = f"{einsum_name}<SEP>{MAPPING_COLUMN}"

0 commit comments

Comments
 (0)