removenoninst.scala

object RemoveNonInst {
  def run(c: Circuit): Circuit = c map onModule
  def onModule(m: DefModule): DefModule = m map onStmt map firrtl.Utils.squashEmpty
  def onStmt(s: Statement): Statement = s map onStmt match {
    case d: WDefInstance => d
    case d: DefInstance => d
    case b: Block => b
    case c: Conditionally => Block(Seq(c.conseq, c.alt))
    case other => EmptyStmt
  }
}