@@ -47,19 +47,25 @@ private[iotesters] object getTopModule {
4747private [iotesters] object getChiselNodes {
4848 import chisel3 .internal .firrtl ._
4949 def apply (circuit : Circuit ): Seq [InstanceId ] = {
50- circuit.components flatMap (_.commands flatMap {
51- case x : DefReg => flatten(x.id)
52- case x : DefRegInit => flatten(x.id)
53- case mem : DefMemory => mem.t match {
54- case _ : Element => Seq (mem.id)
55- case _ => Nil // Do not supoort aggregate type memories
56- }
57- case mem : DefSeqMemory => mem.t match {
58- case _ : Element => Seq (mem.id)
59- case _ => Nil // Do not supoort aggregate type memories
60- }
50+ circuit.components flatMap (_ match {
51+ case m : DefModule =>
52+ m.commands flatMap {
53+ case x : DefReg => flatten(x.id)
54+ case x : DefRegInit => flatten(x.id)
55+ case mem : DefMemory => mem.t match {
56+ case _ : Element => Seq (mem.id)
57+ case _ => Nil // Do not supoort aggregate type memories
58+ }
59+ case mem : DefSeqMemory => mem.t match {
60+ case _ : Element => Seq (mem.id)
61+ case _ => Nil // Do not supoort aggregate type memories
62+ }
63+ case _ => Nil
64+ }
65+ // If it's anything else (i.e., a DefBlackBox), we don't know what to do with it.
6166 case _ => Nil
62- }) filterNot (x => (x.instanceName slice (0 , 2 )) == " T_" )
67+ }
68+ ) filterNot (x => (x.instanceName slice (0 , 2 )) == " T_" )
6369 }
6470}
6571
0 commit comments