Enable @chiselName on non-module classes (#1209)

2 files changed, 4 insertions, 0 deletions

diff --git a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
index e03694a9..b5f617f0 100644
--- a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
+++ b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
@@ -440,6 +440,8 @@ object DynamicNamingStack {
     }
     prefixRef
   }
+  
+  def length() : Int = Builder.namingStackOption.get.length
 }
 
 /** Casts BigInt to Int, issuing an error when the input isn't representable. */
diff --git a/chiselFrontend/src/main/scala/chisel3/internal/Namer.scala b/chiselFrontend/src/main/scala/chisel3/internal/Namer.scala
index e6c8d473..353e51fd 100644
--- a/chiselFrontend/src/main/scala/chisel3/internal/Namer.scala
+++ b/chiselFrontend/src/main/scala/chisel3/internal/Namer.scala
@@ -147,4 +147,6 @@ class NamingStack {
       namingStack.top.addDescendant(prefixRef, until)
     }
   }
+  
+  def length() : Int = namingStack.length
 }