aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/firrtl/passes/memlib/DumpMemoryAnnotations.scala
blob: f88e7e396e7d05514413a41f82f1aabcd6e030a6 (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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
// SPDX-License-Identifier: Apache-2.0

package firrtl
package passes
package memlib

import firrtl.annotations.{CircuitName, ModuleName}
import firrtl.stage.Forms
import firrtl.transforms.BlackBoxInlineAnno

class DumpMemoryAnnotations extends Transform with DependencyAPIMigration {

  override def prerequisites = Forms.MidForm
  override def optionalPrerequisites = Seq.empty
  override def optionalPrerequisiteOf = Forms.MidEmitters
  override def invalidates(a: Transform) = false

  private def vlsiMemGen(annotatedMemory: DefAnnotatedMemory, genBlackBox: Boolean): (String, String) = {
    // MaskedWritePort => WritePort with masked = true
    case class Port(prefix: String, portType: MemPort)

    val masked = annotatedMemory.maskGran.isDefined

    val name = annotatedMemory.name
    val width = {
      val width = bitWidth(annotatedMemory.dataType)
      require(width <= Int.MaxValue)
      width.toInt
    }
    val depth = annotatedMemory.depth.toInt
    val maskGran = if (masked) {
      val maskGran = annotatedMemory.maskGran.get
      require(maskGran <= Int.MaxValue)
      maskGran.toInt
    } else width
    val maskSeg = width / maskGran
    val ports = annotatedMemory.readers.indices.map(number => Port(s"R${number}_", ReadPort)) ++
      annotatedMemory.writers.indices.map(number => Port(s"W${number}_", WritePort)) ++
      annotatedMemory.readwriters.indices.map(number => Port(s"RW${number}_", ReadWritePort))

    val addrWidth = math.max(math.ceil(math.log(depth) / math.log(2)).toInt, 1)
    val readPorts = ports.filter(port => port.portType == ReadPort || port.portType == ReadWritePort)

    val portSpec = ports.flatMap(port =>
      Seq(s"input ${port.prefix}clk", s"input [${addrWidth - 1}:0] ${port.prefix}addr", s"input ${port.prefix}en") ++
        ((port, masked) match {
          case (Port(prefix, ReadPort), _)      => Seq(s"output [${width - 1}:0] ${prefix}data")
          case (Port(prefix, WritePort), false) => Seq(s"input [${width - 1}:0] ${prefix}data")
          case (Port(prefix, WritePort), true) =>
            Seq(s"input [${width - 1}:0] ${prefix}data", s"input [${maskSeg - 1}:0] ${prefix}mask")
          case (Port(prefix, ReadWritePort), false) =>
            Seq(
              s"input ${prefix}wmode",
              s"input [${width - 1}:0] ${prefix}wdata",
              s"output [${width - 1}:0] ${prefix}rdata"
            )
          case (Port(prefix, ReadWritePort), true) =>
            Seq(
              s"input ${prefix}wmode",
              s"input [${maskSeg - 1}:0] ${prefix}wmask",
              s"input [${width - 1}:0] ${prefix}wdata",
              s"output [${width - 1}:0] ${prefix}rdata"
            )
        })
    )

    val decl = readPorts.flatMap(port =>
      Seq(s"reg reg_${port.prefix}ren;", s"reg [${addrWidth - 1}:0] reg_${port.prefix}addr;")
    ) ++ Seq(
      s"reg [${width - 1}:0] ram [${depth - 1}:0];",
      "`ifdef RANDOMIZE_MEM_INIT",
      "  integer initvar;",
      "  initial begin",
      "    #`RANDOMIZE_DELAY begin end",
      s"    for (initvar = 0; initvar < $depth; initvar = initvar+1)",
      s"      ram[initvar] = {${(width - 1) / 32 + 1} {$$random}};"
    ) ++ readPorts.map(port => s"    reg_${port.prefix}addr = {${(addrWidth - 1) / 32 + 1} {$$random}};") ++ Seq(
      "  end",
      "`endif"
    )

    val sequential = {
      def genReadSequential(en: String, prefix: String): Seq[String] = Seq(
        s"always @(posedge ${prefix}clk)",
        s"  reg_${prefix}ren <= $en;",
        s"always @(posedge ${prefix}clk)",
        s"  if ($en) reg_${prefix}addr <= ${prefix}addr;"
      )
      def genWriteSequential(en: String, prefix: String, inputData: String, maskName: String): Seq[String] = Seq(
        s"always @(posedge ${prefix}clk)",
        s"  if ($en) begin"
      ) ++ (0 until maskSeg).map { i =>
        val ifMask = if (masked) s"if (${prefix}${maskName}[$i]) " else ""
        val ram_range = s"${(i + 1) * maskGran - 1}:${i * maskGran}"
        s"    ${ifMask}ram[${prefix}addr][$ram_range] <= ${prefix}$inputData[$ram_range];"
      } ++ Seq("  end")
      ports.flatMap(port =>
        port.portType match {
          case ReadPort  => genReadSequential(port.prefix + "en", port.prefix)
          case WritePort => genWriteSequential(port.prefix + "en", port.prefix, "data", "mask")
          case ReadWritePort =>
            genReadSequential(s"${port.prefix}en && !${port.prefix}wmode", port.prefix) ++
              genWriteSequential(s"${port.prefix}en && ${port.prefix}wmode", port.prefix, "wdata", "wmask")
        }
      )
    }

    val combinational = readPorts.flatMap { port =>
      val data = port.prefix + (if (port.portType == ReadWritePort) "rdata" else "data")
      Seq(
        "`ifdef RANDOMIZE_GARBAGE_ASSIGN",
        s"reg [${((width - 1) / 32 + 1) * 32 - 1}:0] ${port.prefix}random;",
        "`ifdef RANDOMIZE_MEM_INIT",
        "  initial begin",
        "    #`RANDOMIZE_DELAY begin end",
        s"    ${port.prefix}random = {${Seq.fill((width - 1) / 32 + 1)("$random").mkString(", ")}};",
        s"    reg_${port.prefix}ren = ${port.prefix}random[0];",
        "  end",
        "`endif",
        s"always @(posedge ${port.prefix}clk) ${port.prefix}random <= {${Seq.fill((width - 1) / 32 + 1)("$random").mkString(", ")}};",
        s"assign $data = reg_${port.prefix}ren ? ram[reg_${port.prefix}addr] : ${port.prefix}random[${width - 1}:0];",
        "`else",
        s"assign $data = ram[reg_${port.prefix}addr];",
        "`endif"
      )
    }

    val body = if (genBlackBox) "" else s"""
                                           |  ${decl.mkString("\n  ")}
                                           |  ${sequential.mkString("\n  ")}
                                           |  ${combinational.mkString("\n  ")}""".stripMargin

    (name, s"""// name:$name depth:$depth width:$width masked:$masked maskGran:$maskGran maskSeg:$maskSeg
              |module $name(
              |  ${portSpec.mkString(",\n  ")}
              |);
              |
              |$body
              |
              |endmodule""".stripMargin)
  }

  def execute(state: CircuitState): CircuitState = {
    state.copy(annotations = state.annotations.flatMap {
      // convert and remove AnnotatedMemoriesAnnotation to CustomFileEmission
      case AnnotatedMemoriesAnnotation(annotatedMemories) =>
        state.annotations.collect {
          case a: MemLibOutConfigFileAnnotation =>
            Seq(a.copy(annotatedMemories = annotatedMemories))
          case GenVerilogMemBehaviorModelAnno(genBlackBox) =>
            annotatedMemories.map(vlsiMemGen(_, genBlackBox)).map {
              case (name, content) =>
                BlackBoxInlineAnno(ModuleName(name, CircuitName(state.circuit.main)), name + ".v", content)
            }
        }.flatten
      case MemLibOutConfigFileAnnotation(_, Nil) => Nil
      case GenVerilogMemBehaviorModelAnno(_)     => Nil
      case a                                     => Seq(a)
    })
  }
}