From 119e6c7bb9096bbd163a439de09d4932303d0140 Mon Sep 17 00:00:00 2001
From: Jim Lawson
Date: Fri, 19 Aug 2016 12:02:10 -0700
Subject: Simplify autioIOWrap code in computePorts(). As a side-effect, handle
 BlackBoxes correctly.

---
 chiselFrontend/src/main/scala/chisel3/core/Module.scala | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

(limited to 'chiselFrontend/src/main/scala/chisel3/core/Module.scala')

diff --git a/chiselFrontend/src/main/scala/chisel3/core/Module.scala b/chiselFrontend/src/main/scala/chisel3/core/Module.scala
index 24dabcbe..19063664 100644
--- a/chiselFrontend/src/main/scala/chisel3/core/Module.scala
+++ b/chiselFrontend/src/main/scala/chisel3/core/Module.scala
@@ -111,17 +111,18 @@ extends HasId {
     ("clk", clock), ("reset", reset), ("io", io)
   )
 
-  private[core] def computePorts: Seq[firrtl.Port] =
-    for((name, port) <- ports) yield {
-      // If we're auto-wrapping IO definitions, do so now.
-      if (compileOptions.autoIOWrap && name == "io" && !ioDefined) {
-        IO(port)
-      }
+  private[core] def computePorts: Seq[firrtl.Port] = {
+    // If we're auto-wrapping IO definitions, do so now.
+    if (compileOptions.autoIOWrap && !ioDefined) {
+      IO(io)
+    }
+    for ((name, port) <- ports) yield {
       // Port definitions need to know input or output at top-level.
       // By FIRRTL semantics, 'flipped' becomes an Input
       val direction = if(Data.isFirrtlFlipped(port)) Direction.Input else Direction.Output
       firrtl.Port(port, direction)
     }
+  }
 
   private[core] def setupInParent(implicit sourceInfo: SourceInfo): this.type = {
     _parent match {
-- 
cgit v1.2.3