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
|
package firrtl
import Utils._
import DebugUtils._
object Primops {
private val mapPrimop2String = Map[Primop, String](
Add -> "add",
Sub -> "sub",
Addw -> "addw",
Subw -> "subw",
Mul -> "mul",
Div -> "div",
Mod -> "mod",
Quo -> "quo",
Rem -> "rem",
Lt -> "lt",
Leq -> "leq",
Gt -> "gt",
Geq -> "geq",
Eq -> "eq",
Neq -> "neq",
Eqv -> "eqv",
Neqv -> "neqv",
Mux -> "mux",
Pad -> "pad",
AsUInt -> "asUInt",
AsSInt -> "asSInt",
Shl -> "shl",
Shr -> "shr",
Dshl -> "dshl",
Dshr -> "dshr",
Cvt -> "cvt",
Neg -> "neg",
Not -> "not",
And -> "and",
Or -> "or",
Xor -> "xor",
Andr -> "andr",
Orr -> "orr",
Xorr -> "xorr",
Cat -> "cat",
Bit -> "bit",
Bits -> "bits"
)
private val mapString2Primop = mapPrimop2String.map(_.swap)
def fromString(op: String): Primop = mapString2Primop(op)
implicit class PrimopImplicits(op: Primop){
def getString(): String = mapPrimop2String(op)
}
// Borrowed from Stanza implementation
def lowerAndTypePrimop(e: DoPrimop)(implicit logger: Logger): DoPrimop = {
def uAnd(op1: Exp, op2: Exp): Type = {
(op1.getType, op2.getType) match {
case (t1: UIntType, t2: UIntType) => UIntType(UnknownWidth)
case (t1: SIntType, t2) => SIntType(UnknownWidth)
case (t1, t2: SIntType) => SIntType(UnknownWidth)
case _ => UnknownType
}
}
def ofType(op: Exp): Type = {
op.getType match {
case t: UIntType => UIntType(UnknownWidth)
case t: SIntType => SIntType(UnknownWidth)
case _ => UnknownType
}
}
logger.debug(s"lowerAndTypePrimop on ${e.op.getClass.getSimpleName}")
val tpe = e.op match {
case Add => uAnd(e.args(0), e.args(1))
case Sub => SIntType(UnknownWidth)
case Addw => uAnd(e.args(0), e.args(1))
case Subw => uAnd(e.args(0), e.args(1))
case Mul => uAnd(e.args(0), e.args(1))
case Div => uAnd(e.args(0), e.args(1))
case Mod => ofType(e.args(0))
case Quo => uAnd(e.args(0), e.args(1))
case Rem => ofType(e.args(1))
case Lt => UIntType(UnknownWidth)
case Leq => UIntType(UnknownWidth)
case Gt => UIntType(UnknownWidth)
case Geq => UIntType(UnknownWidth)
case Eq => UIntType(UnknownWidth)
case Neq => UIntType(UnknownWidth)
case Eqv => UIntType(UnknownWidth)
case Neqv => UIntType(UnknownWidth)
case Mux => ofType(e.args(1))
case Pad => ofType(e.args(0))
case AsUInt => UIntType(UnknownWidth)
case AsSInt => SIntType(UnknownWidth)
case Shl => ofType(e.args(0))
case Shr => ofType(e.args(0))
case Dshl => ofType(e.args(0))
case Dshr => ofType(e.args(0))
case Cvt => SIntType(UnknownWidth)
case Neg => SIntType(UnknownWidth)
case Not => ofType(e.args(0))
case And => ofType(e.args(0))
case Or => ofType(e.args(0))
case Xor => ofType(e.args(0))
case Andr => UIntType(UnknownWidth)
case Orr => UIntType(UnknownWidth)
case Xorr => UIntType(UnknownWidth)
case Cat => UIntType(UnknownWidth)
case Bit => UIntType(UnknownWidth)
case Bits => UIntType(UnknownWidth)
case _ => ???
}
DoPrimop(e.op, e.args, e.consts, tpe)
}
}
|
