aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/firrtl/transforms/Flatten.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/firrtl/transforms/Flatten.scala')
-rw-r--r--src/main/scala/firrtl/transforms/Flatten.scala9
1 files changed, 8 insertions, 1 deletions
diff --git a/src/main/scala/firrtl/transforms/Flatten.scala b/src/main/scala/firrtl/transforms/Flatten.scala
index 1d9f8083..0aa155bd 100644
--- a/src/main/scala/firrtl/transforms/Flatten.scala
+++ b/src/main/scala/firrtl/transforms/Flatten.scala
@@ -132,7 +132,14 @@ class Flatten extends Transform with DependencyAPIMigration {
val (modNames, instNames) = collectAnns(state.circuit, myAnnotations)
// take incoming annotation and produce annotations for InlineInstances, i.e. traverse circuit down to find all instances to inline
val (newc, modsToInline) = duplicateSubCircuitsFromAnno(state.circuit, modNames, instNames)
- inlineTransform.run(newc, modsToInline.toSet, Set.empty[ComponentName], state.annotations)
+ val flattenedState = inlineTransform.run(newc, modsToInline.toSet, Set.empty[ComponentName], state.annotations)
+
+ val cleanedAnnos = flattenedState.annotations.filterNot {
+ case FlattenAnnotation(_) => true
+ case _ => false
+ }
+
+ flattenedState.copy(annotations = cleanedAnnos)
}
}
}