blob: 38e1c1d973d6f1354f07ebdc77fecc45a067ff13 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
package chiselTests
import chisel3._
import chisel3.experimental.{annotate, ChiselMultiAnnotation}
import chisel3.stage.{ChiselGeneratorAnnotation, ChiselStage}
import firrtl.stage.FirrtlCircuitAnnotation
import org.scalatest.freespec.AnyFreeSpec
import org.scalatest.matchers.should.Matchers
import firrtl.transforms.NoDedupAnnotation
import firrtl.transforms.DontTouchAnnotation
class NewAnnotationsSpec extends AnyFreeSpec with Matchers {
class MuchUsedModule extends Module {
val io = IO(new Bundle {
val in = Input(UInt(16.W))
val out = Output(UInt(16.W))
})
io.out := io.in +% 1.U
}
class UsesMuchUsedModule extends Module {
val io = IO(new Bundle {
val in = Input(UInt(16.W))
val out = Output(UInt(16.W))
})
val mod0 = Module(new MuchUsedModule)
val mod1 = Module(new MuchUsedModule)
val mod2 = Module(new MuchUsedModule)
val mod3 = Module(new MuchUsedModule)
mod0.io.in := io.in
mod1.io.in := mod0.io.out
mod2.io.in := mod1.io.out
mod3.io.in := mod2.io.out
io.out := mod3.io.out
// Give two annotations as single element of the seq - ensures previous API works by wrapping into a seq.
annotate(new ChiselMultiAnnotation { def toFirrtl = Seq(new NoDedupAnnotation(mod2.toNamed)) })
annotate(new ChiselMultiAnnotation { def toFirrtl = Seq(new NoDedupAnnotation(mod3.toNamed)) })
// Pass multiple annotations in the same seq - should get emitted out correctly.
annotate(new ChiselMultiAnnotation {
def toFirrtl =
Seq(new DontTouchAnnotation(mod1.io.in.toNamed), new DontTouchAnnotation(mod1.io.out.toNamed))
})
}
val stage = new ChiselStage
"Ensure all annotations continue to be passed / digested correctly with the new API" - {
"NoDedup and DontTouch work as expected" in {
val dutAnnos = stage
.execute(
Array("-X", "low", "--target-dir", "test_run_dir"),
Seq(ChiselGeneratorAnnotation(() => new UsesMuchUsedModule))
)
val dontTouchAnnos = dutAnnos.collect { case DontTouchAnnotation(target) => target.serialize }
val noDedupAnnos = dutAnnos.collect { case NoDedupAnnotation(target) => target.serialize }
require(dontTouchAnnos.size == 2, s"Exactly two DontTouch Annotations expected but got $dontTouchAnnos ")
require(noDedupAnnos.size == 2, s"Exactly two NoDedup Annotations expected but got $noDedupAnnos ")
val dontTouchAnnosCombined = dontTouchAnnos.mkString(",")
val noDedupAnnosCombined = noDedupAnnos.mkString(",")
noDedupAnnosCombined should include("~UsesMuchUsedModule|MuchUsedModule_2")
noDedupAnnosCombined should include("~UsesMuchUsedModule|MuchUsedModule_3")
dontTouchAnnosCombined should include("~UsesMuchUsedModule|UsesMuchUsedModule/mod1:MuchUsedModule>io_out")
dontTouchAnnosCombined should include("~UsesMuchUsedModule|UsesMuchUsedModule/mod1:MuchUsedModule>io_in")
}
}
}
|
