Make DeadCodeElimination a RegisteredTransform

Signed-off-by: Schuyler Eldridge <schuyler.eldridge@ibm.com>

2 files changed, 10 insertions, 1 deletions

diff --git a/src/main/resources/META-INF/services/firrtl.options.RegisteredTransform b/src/main/resources/META-INF/services/firrtl.options.RegisteredTransform
new file mode 100644
index 00000000..6a6a858d
--- /dev/null
+++ b/src/main/resources/META-INF/services/firrtl.options.RegisteredTransform
@@ -0,0 +1 @@
+firrtl.transforms.DeadCodeElimination
diff --git a/src/main/scala/firrtl/transforms/DeadCodeElimination.scala b/src/main/scala/firrtl/transforms/DeadCodeElimination.scala
index 523c997b..8f9fad46 100644
--- a/src/main/scala/firrtl/transforms/DeadCodeElimination.scala
+++ b/src/main/scala/firrtl/transforms/DeadCodeElimination.scala
@@ -11,6 +11,8 @@ import firrtl.Mappers._
 import firrtl.WrappedExpression._
 import firrtl.Utils.{throwInternalError, toWrappedExpression, kind}
 import firrtl.MemoizedHash._
+import firrtl.options.RegisteredTransform
+import scopt.OptionParser
 
 import collection.mutable
 import java.io.{File, FileWriter}
@@ -30,10 +32,16 @@ import java.io.{File, FileWriter}
   * circumstances of their instantiation in their parent module, they will still not be removed. To
   * remove such modules, use the [[NoDedupAnnotation]] to prevent deduplication.
   */
-class DeadCodeElimination extends Transform with ResolvedAnnotationPaths {
+class DeadCodeElimination extends Transform with ResolvedAnnotationPaths with RegisteredTransform {
   def inputForm = LowForm
   def outputForm = LowForm
 
+  def addOptions(parser: OptionParser[AnnotationSeq]): Unit = parser
+    .opt[Unit]("no-dce")
+    .action( (x, c) => c :+ NoDCEAnnotation )
+    .maxOccurs(1)
+    .text("Do NOT run dead code elimination")
+
   /** Based on LogicNode ins CheckCombLoops, currently kind of faking it */
   private type LogicNode = MemoizedHash[WrappedExpression]
   private object LogicNode {