From 1705980cc447e698ca431c4eca2c91bf73a2aab1 Mon Sep 17 00:00:00 2001
From: Albert Magyar
Date: Thu, 14 May 2020 13:17:09 -0700
Subject: Make find_port return Unit and use Foreachers in ResolveKinds

---
 src/main/scala/firrtl/passes/ResolveKinds.scala | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/main/scala/firrtl/passes/ResolveKinds.scala b/src/main/scala/firrtl/passes/ResolveKinds.scala
index 0f2da3fa..809ba2fc 100644
--- a/src/main/scala/firrtl/passes/ResolveKinds.scala
+++ b/src/main/scala/firrtl/passes/ResolveKinds.scala
@@ -5,6 +5,7 @@ package firrtl.passes
 import firrtl._
 import firrtl.ir._
 import firrtl.Mappers._
+import firrtl.traversals.Foreachers._
 import firrtl.options.PreservesAll
 
 object ResolveKinds extends Pass with PreservesAll[Transform] {
@@ -13,8 +14,8 @@ object ResolveKinds extends Pass with PreservesAll[Transform] {
 
   type KindMap = collection.mutable.HashMap[String, Kind]
 
-  def find_port(kinds: KindMap)(p: Port): Port = {
-    kinds(p.name) = PortKind ; p
+  private def find_port(kinds: KindMap)(p: Port): Unit = {
+    kinds(p.name) = PortKind
   }
 
   def resolve_expr(kinds: KindMap)(e: Expression): Expression = e match {
@@ -37,7 +38,7 @@ object ResolveKinds extends Pass with PreservesAll[Transform] {
 
   def resolve_kinds(m: DefModule): DefModule = {
     val kinds = new KindMap
-    m.map(find_port(kinds))
+    m.foreach(find_port(kinds))
     m.map(resolve_stmt(kinds))
   }
 
-- 
cgit v1.2.3