aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorAdam Izraelevitz2015-07-28 13:29:55 -0700
committerAdam Izraelevitz2015-07-28 13:29:55 -0700
commit0c7aca561aef907314b0d9c9737fcea04ae6ce82 (patch)
tree1b5a4da96413bd34d6b5e05ed2cbb2312d1a4b1d /src
parent69e4bc2c4f4e7c2940633f12a40a5550205eb75e (diff)
Integrated bigint. Mostly works, but getting "cast" error for make Test.
Diffstat (limited to 'src')
-rw-r--r--src/main/stanza/bigint.stanza424
-rw-r--r--src/main/stanza/custom-passes.stanza7
-rw-r--r--src/main/stanza/errors.stanza3
-rw-r--r--src/main/stanza/firrtl-ir.stanza7
-rw-r--r--src/main/stanza/firrtl-lexer.stanza12
-rw-r--r--src/main/stanza/firrtl-test-main.stanza1
-rw-r--r--src/main/stanza/flo.stanza6
-rw-r--r--src/main/stanza/ir-parser.stanza26
-rw-r--r--src/main/stanza/passes.stanza17
-rw-r--r--src/main/stanza/verilog.stanza4
10 files changed, 472 insertions, 35 deletions
diff --git a/src/main/stanza/bigint.stanza b/src/main/stanza/bigint.stanza
new file mode 100644
index 00000000..f5122323
--- /dev/null
+++ b/src/main/stanza/bigint.stanza
@@ -0,0 +1,424 @@
+defpackage bigint :
+ import core
+ import verse
+
+
+public defn to-hex (num: Int) -> String :
+ val buf = StringBuffer()
+ print(buf, "0x")
+ var rems = List<Int>()
+ var x : Int = num
+ defn* loop () :
+ rems = List(x % 16, rems)
+ x = x / 16
+ if x > 0: loop()
+ loop()
+ for r in rems do :
+ print(buf, to-char((to-int('0') + r) when r < 10 else (to-int('a') + r - 10)))
+ to-string(buf)
+
+
+val val-n-bytes = 4
+val val-n-bits = val-n-bytes * 8
+val val-all-ones = -1 ;; TODO
+val val-n-half-bits = 32 / 2
+defn val-n-words (nbits: Int) -> Int : 1 + (nbits - 1) / 32 ;val-n-bits
+defn val-n-half-words (nbits: Int) -> Int : (1 + (nbits - 1)) / val-n-half-bits
+defn val-top-bit (v: Int) -> Int : v >> (32 - 1)
+defn val-n-full-words (nbits: Int) -> Int : nbits / 32
+defn val-n-word-bits (nbits: Int) -> Int : nbits % 32
+defn mask-val (n: Int) : val-all-ones >> (32 - n)
+
+public defclass BigInt <: Gettable & Settable & Lengthable
+public defmulti to-bin (b: BigInt) -> String
+public defmulti num-words (b: BigInt) -> Int
+public defmulti dat (b: BigInt) -> Array<Int>
+public defn BigInt (len: Int) :
+ val d = Array<Int>(val-n-words(len))
+ new BigInt :
+ defmethod get (this, i: Int) -> Int : d[i]
+ defmethod set (this, i: Int, x: Int) : d[i] = x; & (-1) ;val-all-ones
+ defmethod dat (this) -> Array<Int> : d
+ defmethod length (this) -> Int : len
+ defmethod num-words (this) -> Int : length(d)
+ defmethod to-bin (this) -> String :
+ string-join $
+ generate<Char> :
+ defn* loop (pos:Int) :
+ if (pos >= 0) :
+ yield(if (d[pos / 32] >> (pos % 32))&1 == 1: '1' else: '0')
+ loop(pos - 1)
+ loop(len - 1)
+ defmethod to-string (this) :
+ var s = ""
+ for x in d do : s = string-join([x s])
+ s
+ ;defmethod to-string (this) : string-join(["BigInt<" len ">(" to-bin(this) ")"])
+
+defmethod print (o:OutputStream, b:BigInt) :
+ print(o, to-string(b))
+
+defn assert (cond:True|False, msg:String) : if not cond: error(msg)
+
+defn assert (cond:True|False) : assert(cond, "failure")
+
+;; loop macro with starting values
+
+;; defn map-range (s:Int, e:Int, f: (Int) -> False) :
+;; for i in s to e do : f(i)
+
+defn map! (f: (Int, Int) -> Int, d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+ ;; assert(length(d) == length(s0) and length(s0) == length(s1),
+ ;; string-join(["LENS != " num-words(d) " " num-words(s0) " " num-words(s1)]))
+ for i in 0 to num-words(d) do :
+ d[i] = f(s0[i], s1[i])
+ d
+
+defn mask! (d: BigInt) -> BigInt :
+ val n-full-words = val-n-full-words(length(d))
+ val n-word-bits = val-n-word-bits(length(d))
+ for i in 0 to n-full-words do :
+ d[i] = val-all-ones
+ for i in n-full-words to num-words(d) do :
+ d[i] = 0
+ if n-word-bits > 0 :
+ d[n-full-words] = mask-val(n-word-bits)
+ d
+
+defn trim! (d:BigInt) -> BigInt :
+ val n-full-words = val-n-full-words(length(d))
+ val n-word-bits = val-n-word-bits(length(d))
+ if n-word-bits > 0 :
+ d[n-full-words] = d[n-full-words] & mask-val(n-word-bits)
+ d
+
+defn extract! (d:BigInt, s0:BigInt, e:Int, s:Int) -> BigInt : trim!(rsh!(d, s0, s))
+
+public defn bits (s0:BigInt, e:Int, s:Int) -> BigInt : extract!(BigInt(e - s + 1), s0, e, s)
+
+public defn bit (s0:BigInt, s:Int) -> Int :
+ val wi = s / 32 ;val-n-bits
+ val bi = s % 32 ;val-n-bits
+ s0[wi] >> bi
+
+defn inject! (d:BigInt, s0:BigInt, f:BigInt, start:Int) -> BigInt :
+ val bw = length(f)
+ val msk = mask!(BigInt(bw))
+ val msk-lsh = msk << start
+ val inv-msk-lsh = bit-invert!(msk-lsh, msk-lsh)
+ val f-lsh = f << start
+ (s0 & inv-msk-lsh) | f-lsh
+
+defn map! (f: (Int) -> Int, d: BigInt, s0: BigInt) -> BigInt :
+ ;; assert(length(d) == length(s0), string-join(["LENS != " num-words(d) " " num-words(s0)]))
+ for i in 0 to num-words(d) do :
+ d[i] = f(s0[i])
+ d
+
+defn fill! (f: (Int) -> Int, b: BigInt) :
+ for i in 0 to num-words(b) do :
+ b[i] = f(i)
+ b
+
+public defmethod equal? (x:BigInt, y:BigInt) -> True|False :
+ if num-words(x) != num-words(y) :
+ false
+ else :
+ var eq? = true
+ for i in 0 to num-words(x) do :
+ val e = (x[i] == y[i])
+ if not e : println-all(["NOT-EQUAL " x " AND " y " i=" i " X[i] " x[i] " Y[i] " y[i]])
+ eq? = eq? and e
+ eq?
+
+public defn less?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False :
+ sub!(diff, x, y)
+ val-top-bit(diff[num-words(diff) - 1]) == 1
+
+public defn less? (x:BigInt, y:BigInt) -> True|False : op(less?!, x, y)
+
+public defn less-eq?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False :
+ sub!(diff, y, x)
+ val-top-bit(diff[num-words(diff) - 1]) == 0
+
+public defn less-eq? (x:BigInt, y:BigInt) -> True|False : op(less-eq?!, x, y)
+
+public defn greater?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False : not less-eq?!(diff, x, y)
+
+public defn greater? (x:BigInt, y:BigInt) -> True|False : op(greater?!, x, y)
+
+public defn greater-eq?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False : not less?!(diff, x, y)
+
+public defn greater-eq? (x:BigInt, y:BigInt) -> True|False : op(greater-eq?!, x, y)
+
+public defn BigIntLit (f: (Int) -> Int, len: Int) :
+ fill!(f, BigInt(len))
+
+;val hex-nibbles = "0123456789abcdef"
+
+defn as-digit (c: Char) -> Int :
+ var index = 0
+ for (x in "0123456789abcdef", i in 0 to false) do :
+ if x == c: index = i
+ index
+ ;val x = index-of(hex-nibbles, c)
+ ;x as Int
+
+public defn BigIntLit (s: String) : BigIntLit(s, -1)
+
+;public defn BigIntLit (i: Int) : BigIntLit(string-join(["x" to-hex(to-string(i))]))
+
+public defn BigIntLit (s: String, w:Int) :
+ val hd = s[0]
+ val base = if hd == 'b': 1 else if hd == 'x': 4 else: 'd'
+ val shamt = if base == 'b': 1 else if base == 'x': 4 else: 4
+ val digits = if base == 'b': substring(s, 1) else if base == 'x': substring(s, 1) else: s
+ val len = if w == -1: length(digits) * shamt else: w
+ println-all(["STRING " s])
+ println-all(["BASE " base " SHAMT " shamt " DIGITS " digits " LEN " len])
+ var lit = BigInt(len)
+ for i in 0 to num-words(lit) do :
+ lit[i] = 0
+
+
+ switch {base == _} :
+ 'd' :
+ for i in 0 to length(digits) do :
+ val y = shift-left(lit,3) + shift-left(lit,1)
+ val x = to-int(substring(digits,i,i + 1))
+ val z = BigInt(length(y))
+ z[0] = x
+ lit = y + z
+ println-all(["DIGIT = " x])
+ println-all(["Before LIT = " lit])
+
+ ;println-all(["After LIT = " lit])
+ ;println-all(["RES = " to-bin(lit)])
+ ;println-all(["RES = " lit])
+ lit
+ else :
+ for i in 0 to length(digits) do :
+ val off = (length(digits) - 1 - i) * shamt
+ val wi = off / 32;val-n-bits
+ val bi = off % 32;val-n-bits
+ ;println-all(["OFF " off " wi " wi " bi " bi " digits " digits])
+ lit[wi] = lit[wi] | ((as-digit(digits[i]) & 1) << bi)
+ ;println-all([" lit[wi] " lit[wi] " => " lit])
+ ;println-all(["RES = " to-bin(lit)])
+ ;println-all(["RES = " lit])
+ lit
+
+public defn sizeof (in: Int) -> Int :
+ max(1, ceil-log2(in + 1))
+
+public defn BigIntLit (x: Int, w: Int) :
+ fill!({_ + x}, BigInt(w))
+
+public defn BigIntLit (x: Int) : BigIntLit(x, sizeof(x))
+
+defn op (f:(BigInt, BigInt, BigInt) -> True|False, x:BigInt, y:BigInt) -> True|False :
+ f(BigInt(max(length(x), length(y))), x, y)
+
+defn op (f:(BigInt, BigInt, BigInt) -> BigInt, x:BigInt, y:BigInt) -> BigInt :
+ f(BigInt(max(length(x), length(y))), x, y)
+
+defn op (f:(BigInt, BigInt) -> BigInt, x:BigInt) -> BigInt :
+ f(BigInt(length(x)), x)
+
+defn int (x: True|False) -> Int : if x : 1 else : 0
+
+public defn plus! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+ var carry = false
+ for i in 0 to num-words(d) do :
+ d[i] = s0[i] + s1[i] + int(carry)
+ carry = ((s0[i] + s1[i]) < s0[i]) or (d[i] < int(carry))
+ d
+
+public defn plus (x:BigInt, y:BigInt) -> BigInt : op(plus!, x, y)
+
+public defn cat! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+ lsh!(d, s0, length(s1))
+ ;; println-all([" LSH! " s0 " " length(s1) " => " d])
+ for i in 0 to num-words(s1) do :
+ d[i] = d[i] | s1[i]
+ d
+
+public defn cat (x:BigInt, y:BigInt) -> BigInt :
+ ;; println-all(["> CAT " x " " y])
+ val res = cat!(BigInt(length(x) + length(y)), x, y)
+ ;; println-all(["< CAT " x " " y " => " res])
+ res
+
+public defn cat (args:Streamable<BigInt>) -> BigInt :
+ reduce(cat, args)
+
+public defn sub! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+ var borrow = false
+ for i in 0 to num-words(d) do :
+ d[i] = s0[i] - s1[i] + int(borrow)
+ borrow = (s0[i] < (s0[i] - s1[i])) or (s0[i] - s1[i]) < d[i]
+ d
+
+public defn sub (x:BigInt, y:BigInt) -> BigInt : op(sub!, x, y)
+
+public defn neg! (d: BigInt, s0: BigInt) -> BigInt :
+ var borrow = false
+ for i in 0 to num-words(d) do :
+ d[i] = neg(s0[i]) - int(borrow)
+ borrow = (s0[i] > 0) or (d[i] > 0)
+ d
+
+public defn neg (x:BigInt) -> BigInt : op(neg!, x)
+
+public defn rsha! (d:BigInt, s0:BigInt, amount:Int) -> BigInt :
+ val w = length(s0)
+ val nw = num-words(d)
+ val n-shift-bits = amount % 32;val-n-bits
+ val n-shift-words = amount / 32;val-n-bits
+ val n-rev-shift-bits = 32 - n-shift-bits;val-n-bits
+ val is-zero-carry = n-shift-bits == 0
+ val msb = s0[nw - 1] >> (w - nw * 32 - 1);val-n-bits
+ var carry = 0;
+
+ if msb == 0 :
+ for i in 0 to n-shift-words do :
+ d[nw - i - 1] = 0
+
+ defn* loopy (i:Int) :
+ if i >= n-shift-words :
+ val x = s0[i]
+ d[i - n-shift-words] = (x >> n-shift-bits) | carry
+ carry = if is-zero-carry: 0 else: x << n-rev-shift-bits
+ loopy(i - 1)
+ loopy(nw - 1)
+
+ if msb != 0 :
+ val boundary = (w - amount)
+
+ defn* loop (i:Int) :
+ if i >= 0 :
+ val idx = i * val-n-bits
+ if idx > boundary :
+ d[i] = val-all-ones
+ loop(i - 1)
+ else :
+ d[i] = d[i] | (val-all-ones << (boundary - idx))
+ d[nw - 1] = d[nw - 1] & (val-all-ones >> ((nw - 1) * val-n-bits - w))
+ loop(nw - 1)
+ d
+
+public defn signed-shift-right (b:BigInt, n:Int) -> BigInt : rsha!(BigInt(length(b)), b, n)
+
+public defn rsh! (d:BigInt, s0:BigInt, amount:Int) -> BigInt :
+ val nw = num-words(d)
+ var carry = 0
+ val n-shift-bits = amount % val-n-bits
+ val n-shift-words = amount / val-n-bits
+ val n-rev-shift-bits = val-n-bits - n-shift-bits
+ val is-zero-carry = n-shift-bits == 0
+ for i in 0 to n-shift-words do :
+ d[nw - i - 1] = 0
+ defn* loop (i:Int) :
+ if i >= n-shift-words :
+ val x = s0[i]
+ d[i - n-shift-words] = (x >> n-shift-bits) | carry
+ carry = if is-zero-carry: 0 else: x << n-rev-shift-bits
+ loop(i - 1)
+ loop(nw - 1)
+ d
+
+public defn shift-right (b:BigInt, n:Int) -> BigInt : rsh!(BigInt(length(b)), b, n)
+
+public defn lsh! (d:BigInt, s0:BigInt, amount:Int) :
+ ;; println-all(["LSH " s0 " AMOUNT " amount " INTO BIGINT<" length(d) ">"])
+ val n-shift-bits = amount % val-n-bits
+ val n-shift-words = amount / val-n-bits
+ val n-rev-shift-bits = val-n-bits - n-shift-bits
+ val is-zero-carry = n-shift-bits == 0
+ for i in 0 to num-words(d) do :
+ d[i] = 0
+ var carry = 0;
+ ;; println-all(["LSH AMOUNT " amount " VNB " val-n-bits " NSB " n-shift-bits " NSW " n-shift-words " NRSB " n-rev-shift-bits])
+ for i in 0 to (num-words(d) - n-shift-words) do :
+ val x = if i >= num-words(s0) : 0 else: s0[i]
+ ;; println-all([" SHIFTING " (i + n-shift-words) " VAL " (x << n-shift-bits)])
+ d[i + n-shift-words] = (x << n-shift-bits) | carry
+ carry = if is-zero-carry: 0 else: x >> n-rev-shift-bits
+ d
+
+public defn shift-left (b:BigInt, n:Int) -> BigInt : lsh!(BigInt(length(b) + n), b, n)
+
+;; defn mul! (d: BigInt, s0: BigInt, s1: BigInt, nb0: Int, nb1: Int) -> BigInt :
+;; ;; Adapted from Hacker's Delight, from Knuth
+;; var nbd = nb0 + nb1
+;; for i in 0 to val-n-words(nbd) do :
+;; d[i] = 0
+;;
+;; half-val-t* w = reinterpret-cast<half-val-t*>(d);
+;; half-val-t* u = reinterpret-cast<half-val-t*>(s0);
+;; half-val-t* v = reinterpret-cast<half-val-t*>(s1);
+;; val m = val-n-half-words(nb0)
+;; val n = val-n-half-words(nb1)
+;; val p = val-n-half-words(nbd)
+;;
+;; for j in 0 to n do :
+;; var k = 0
+;; for i in 0 to min(m, p - j) do :
+;; val t = u[i] * v[j] + w[i + j] + k
+;; w[i + j] = t
+;; k = t >> val-n-half-bits()
+;; if ((j + m) < p)
+;; w[j + m] = k
+;; d
+
+public defn bit-xor! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-xor, d, s0, s1)
+
+public defn bit-xor (b0:BigInt, b1:BigInt) : op(bit-xor!, b0, b1)
+
+public defn bit-or! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-or, d, s0, s1)
+
+public defn bit-or (b0:BigInt, b1:BigInt) : op(bit-or!, b0, b1)
+
+public defn bit-and! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-and, d, s0, s1)
+
+public defn bit-and (b0:BigInt, b1:BigInt) : op(bit-and!, b0, b1)
+
+public defn bit-invert! (d: BigInt, s0: BigInt) : map!(bit-xor, d, s0, mask!(d))
+
+public defn bit-invert (b0:BigInt) : op(bit-invert!, b0)
+
+defn check (msg:String, x:BigInt, e:BigInt) :
+ println-all([msg " " x " E " e " ? " x == e])
+
+defn check (msg:String, x:BigInt) :
+ println-all([msg " " x])
+
+;; check("Ba ", BigIntLit({ _ }, 1))
+;; check("Bb ", BigIntLit({ _ }, 16))
+;; check("Bc ", BigIntLit({ _ }, 32))
+;; check("Bd ", BigIntLit({ _ }, 48))
+;; check("Be ", BigIntLit({ _ }, 64))
+;; check("Bf ", BigIntLit({ _ }, 65))
+;; check("B1 ", BigIntLit(1, 8))
+;; check("B2 ", BigIntLit(2, 8))
+;; check("B+ ", BigIntLit(3, 8) + BigIntLit(5, 8), BigIntLit(3 + 5, 8))
+;; check("B- ", BigIntLit(5, 8) + BigIntLit(3, 8), BigIntLit(5 + 3, 8))
+;; check("B| ", BigIntLit(5, 8) | BigIntLit(9, 8), BigIntLit(5 | 9, 8))
+;; check("B& ", BigIntLit(5, 8) & BigIntLit(9, 8), BigIntLit(5 & 9, 8))
+;; check("B^ ", BigIntLit(5, 8) ^ BigIntLit(9, 8), BigIntLit(5 ^ 9, 8))
+;; check("B< ", BigIntLit(5, 8) << 1, BigIntLit(5 << 1, 9))
+;; check("B< ", BigIntLit(5, 3) << 10, BigIntLit(5 << 1, 13))
+;; check("B< ", BigIntLit(5, 3) << 32, BigIntLit(5 << 1, 38))
+;; check("B< ", BigIntLit("b1010") << 1, BigIntLit(10 << 1, 5))
+;check("S1 ", BigIntLit("xfafa") << 16, BigIntLit("xfafa0000", 32))
+;;; check("B< ", BigIntLit(5, 3) << 64, BigIntLit(5 << 1, 67))
+;;; check("BN ", neg(BigIntLit(2, 8)), BigIntLit(-2, 8))
+;check("S2 ", BigIntLit("b11111010") << 8, BigIntLit("b1111101000000000", 16))
+;check("C1 ", cat(BigIntLit("b11111010", 8), BigIntLit("b10111100", 8)), BigIntLit("b1111101010111100", 16))
+;check("C3 ", cat(cat(BigIntLit("b1111"), BigIntLit("b1010")), cat(BigIntLit("b1011"), BigIntLit("b1100"))), BigIntLit("b1111101010111100", 16))
+;check("C4 ", cat([BigIntLit("b1111"), BigIntLit("b1010"), BigIntLit("b1011"), BigIntLit("b1100")]), BigIntLit("b1111101010111100", 16))
+;check("C5 ", BigIntLit("b101111001"), BigIntLit("b101111001"))
+;check("C6 ", cat(BigIntLit("b1"), BigIntLit("b01111001")), BigIntLit("b101111001"))
+;check("C7 ", cat(BigIntLit("b11101"), BigIntLit("b101111001")), BigIntLit("b11101101111001"))
+;check("C8 ", cat([BigIntLit("b11"), BigIntLit("b101"), BigIntLit("b1011"), BigIntLit("b11001")]), BigIntLit("b11101101111001"))
+;check("C0 ", bits(BigIntLit("b11101101111001"), 10, 1), BigIntLit("b0110111100"))
diff --git a/src/main/stanza/custom-passes.stanza b/src/main/stanza/custom-passes.stanza
index 4904d6d1..ecd7aed1 100644
--- a/src/main/stanza/custom-passes.stanza
+++ b/src/main/stanza/custom-passes.stanza
@@ -3,6 +3,7 @@ defpackage firrtl/custom-passes :
import verse
import firrtl/ir-utils
import firrtl/ir2
+ import bigint
;============ When Coverage =============
public defstruct WhenCoverage <: Pass :
@@ -46,7 +47,7 @@ defn when-coverage (port-name:Symbol, reg-name:Symbol, instrument?:HashTable<Sym
val sym = HashTable<Symbol,Int>(symbol-hash)
val w1 = IntWidth(1)
val t1 = UIntType(w1)
- val u1 = UIntValue(to-long $ 1,w1)
+ val u1 = UIntValue(BigIntLit("1"),w1)
defn when-coverage (s:Stmt) -> Stmt :
match(s) :
(s:Conditionally) :
@@ -71,7 +72,7 @@ defn when-coverage (port-name:Symbol, reg-name:Symbol, instrument?:HashTable<Sym
if length(w-ls) != 0 :
val reg-ref = Ref(reg-name,UIntType(IntWidth(length(w-ls))))
;add{logic,_} $ DefRegister(FileInfo(),name(reg-ref),type(reg-ref)) TODO add clock and reset
- add{logic,_} $ OnReset(FileInfo(),reg-ref,UIntValue(to-long $ 0,IntWidth(length(w-ls))))
+ add{logic,_} $ OnReset(FileInfo(),reg-ref,UIntValue(BigIntLit("0"),IntWidth(length(w-ls))))
for (x in w-ls, i in 0 to false) do :
add{logic,_} $ DefWire(FileInfo(),name(x),type(x))
add{logic,_} $ Connect(FileInfo(),x,DoPrim(BIT-SELECT-OP,list(reg-ref),list(i),UIntType(w1)))
@@ -81,7 +82,7 @@ defn when-coverage (port-name:Symbol, reg-name:Symbol, instrument?:HashTable<Sym
if length(i-ls) != 0 :
for (x in i-ls, i in 0 to false) do :
add{logic,_} $ DefWire(FileInfo(),name(x),type(x))
- add{logic,_} $ Connect(FileInfo(),x,UIntValue(to-long $ 0,UnknownWidth()))
+ add{logic,_} $ Connect(FileInfo(),x,UIntValue(BigIntLit("0"),UnknownWidth()))
if instrument?[name(m)] : add{logic,_} $ Connect(FileInfo(),port-ref,concat-all(append(w-ls,i-ls)))
diff --git a/src/main/stanza/errors.stanza b/src/main/stanza/errors.stanza
index d8969dd3..80ad3e56 100644
--- a/src/main/stanza/errors.stanza
+++ b/src/main/stanza/errors.stanza
@@ -6,6 +6,7 @@ defpackage firrtl/errors :
import firrtl/primops
import firrtl/passes
import firrtl-main
+ import bigint
; TODO
; make sure it compiles, write tests, look over code to make sure its right
@@ -277,7 +278,7 @@ public defn check-high-form (c:Circuit) -> Circuit :
(e) : add(errors,InvalidIndex(info))
(e:DoPrim) : check-high-form-primop(e,errors,info)
(e:UIntValue) :
- if value(e) < to-long $ 0 : add(errors,NegUInt(info))
+ if value(e) < BigIntLit("0") : add(errors,NegUInt(info))
(e) : false
map(check-high-form-w{info,_:Width},e)
map(check-high-form-t{info,_:Type},e)
diff --git a/src/main/stanza/firrtl-ir.stanza b/src/main/stanza/firrtl-ir.stanza
index 15666a9a..822b8610 100644
--- a/src/main/stanza/firrtl-ir.stanza
+++ b/src/main/stanza/firrtl-ir.stanza
@@ -1,6 +1,7 @@
defpackage firrtl/ir2 :
import core
import verse
+ import bigint
public defmulti info! (x:?) -> FileInfo
public defmethod info! (x:?) : FileInfo()
@@ -31,7 +32,7 @@ public defstruct UnknownWidth <: Width
public defstruct IntWidth <: Width :
width: Int
public defstruct LongWidth <: Width :
- width : Long
+ width: Long
public definterface PrimOp
public val ADD-OP = new PrimOp
@@ -85,10 +86,10 @@ public defstruct Index <: Expression :
value: Int
type: Type with: (as-method => true)
public defstruct UIntValue <: Expression :
- value: Long
+ value: BigInt
width: Width
public defstruct SIntValue <: Expression :
- value: Long
+ value: BigInt
width: Width
public defstruct DoPrim <: Expression :
op: PrimOp
diff --git a/src/main/stanza/firrtl-lexer.stanza b/src/main/stanza/firrtl-lexer.stanza
index f0673a72..8ee90f68 100644
--- a/src/main/stanza/firrtl-lexer.stanza
+++ b/src/main/stanza/firrtl-lexer.stanza
@@ -2,6 +2,7 @@ defpackage firrtl/lexer :
import core
import core/stringeater
import verse
+ import bigint
;=============== PUBLIC INTERFACE ===========================
public defn lex (text:String) -> List<Token> :
@@ -287,11 +288,12 @@ defn eat-number () :
(f:Float) : token-eaten(Token(f, info))
(f:False) : throw(InvalidNumber(info))
else :
- match(to-long(str)) :
- (l:Long) :
- if l < (to-long("2147483647") as Long) and l > (to-long("-2147483648") as Long) : token-eaten(Token(to-int(str), info))
- else : token-eaten(Token(l, info))
- (l:False) : token-eaten(Token(to-int(str), info))
+ token-eaten(Token(BigIntLit(str),info))
+ ;match(to-long(str)) :
+ ; (l:Long) :
+ ; if l < (to-long("2147483647") as Long) and l > (to-long("-2147483648") as Long) : token-eaten(Token(to-int(str), info))
+ ; else : token-eaten(Token(l, info))
+ ; (l:False) : token-eaten(Token(to-int(str), info))
;else : token-eaten(Token(to-int(str), info)) <- ADAM CHANGE, FROM THIS
defn eat-here-string () :
diff --git a/src/main/stanza/firrtl-test-main.stanza b/src/main/stanza/firrtl-test-main.stanza
index beb288b1..a5050c47 100644
--- a/src/main/stanza/firrtl-test-main.stanza
+++ b/src/main/stanza/firrtl-test-main.stanza
@@ -13,6 +13,7 @@
#include("compilers.stanza")
#include("flo.stanza")
#include("verilog.stanza")
+#include("bigint.stanza")
;Custom Packages
#include("custom-passes.stanza")
diff --git a/src/main/stanza/flo.stanza b/src/main/stanza/flo.stanza
index d8fe705f..f4b453ba 100644
--- a/src/main/stanza/flo.stanza
+++ b/src/main/stanza/flo.stanza
@@ -4,6 +4,7 @@ defpackage firrtl/flo :
import firrtl/ir-utils
import firrtl/ir2
import firrtl/passes
+ import bigint
;============= Flo Backend ================
@@ -72,9 +73,6 @@ defn prim-width (type:Type) -> Int :
(t:ClockType) : 1
(t) : error("Bad prim width type")
-defn sizeof (in: Int) -> Int :
- max(1, ceil-log2(in + 1))
-
defn emit-all (es:Streamable, top:Symbol) :
for e in es do :
match(e) :
@@ -171,7 +169,7 @@ defn emit-s (s:Stmt, flokinds:HashTable<Symbol,FloKind>, top:Symbol,sh:HashTable
(s:DefNode) :
emit-all([top "::" name(s) " = " maybe-mov(value(s)) value(s) "\n"], top)
(s:Begin) : do(emit-s{_, flokinds, top,sh}, body(s))
- (s:Connect) : emit-connect(s,UIntValue(to-long(1),IntWidth(1)))
+ (s:Connect) : emit-connect(s,UIntValue(BigIntLit("1"),IntWidth(1)))
(s:Conditionally) : emit-connect(conseq(s) as Connect,pred(s))
(s) : s
diff --git a/src/main/stanza/ir-parser.stanza b/src/main/stanza/ir-parser.stanza
index 7e07eb8b..44b51f30 100644
--- a/src/main/stanza/ir-parser.stanza
+++ b/src/main/stanza/ir-parser.stanza
@@ -4,6 +4,7 @@ defpackage firrtl/parser :
import firrtl/ir2
import stz/parser
import firrtl/lexer
+ import bigint
;======= Convenience Functions ========
defn first-info? (form) -> FileInfo|False :
@@ -88,6 +89,9 @@ defsyntax firrtl :
;Parses next form if symbol
sym = (?x) when unwrap-token(x) typeof Symbol :
unwrap-token(x)
+
+ bigint = (?x) when unwrap-token(x) typeof BigInt :
+ unwrap-token(x)
;Error Handling Productions
defrule :
@@ -107,6 +111,10 @@ defsyntax firrtl :
=! = (=) : `=
=! != () : FPE(form, "Expected a '=' here.")
+ ;Error if not a single bigint
+ bigint$ = (?i:#bigint ?rest ...) when empty?(rest) : i
+ bigint$ != () : FPE(form, "Expected a single big integer literal here.")
+
;Error if not a single integer
int$ = (?i:#int ?rest ...) when empty?(rest) : i
int$ != () : FPE(form, "Expected a single integer literal here.")
@@ -191,7 +199,7 @@ defsyntax firrtl :
type = (?t:#typeterm ?ops:#typeop ...) : apply-suffix-ops(t, ops)
type = (?t:#clktype) : t
- typeop = ((@get ?size:#int$)) : (fn (t) : VectorType(t, size))
+ typeop = ((@get ?size:#bigint$)) : (fn (t) : VectorType(t, size[0]))
typeterm = (?t:#inttype) : t
typeterm = ({?fs:#field ... ?rest ...}) :
@@ -210,10 +218,10 @@ defsyntax firrtl :
accdir = (rdwr) : RDWR
defrule width :
- width = (?x:#intorlong) :
- match(x) :
- (x:Int) : IntWidth(x)
- (x:Long) : LongWidth(x)
+ width = (?x:#bigint) :
+ if num-words(x) == 1 : IntWidth(x[0])
+ else if num-words(x) == 2 : LongWidth(to-long(x[1]) * to-long(1 << 32) + to-long(x[0]))
+ else : IntWidth(x[0])
width = (?) : UnknownWidth()
;Main Statement Productions
@@ -253,20 +261,20 @@ defsyntax firrtl :
defrule exp :
;Suffix Operators
exp = (?x:#expterm ?ops:#expop ...) : apply-suffix-ops(x, ops)
- expop = ((@get ?f:#int)) : (fn (x) : Index(x, f, UnknownType()))
+ expop = ((@get ?f:#bigint)) : (fn (x) : Index(x, f[0], UnknownType()))
expop = (. ?f:#id!) : (fn (x) : Subfield(x, f, UnknownType()))
;Prefix Operators
- expterm = (?t:#inttype(?v:#long$)) :
+ expterm = (?t:#inttype(?v:#bigint$)) :
match(t) :
(t:UIntType) : UIntValue(v, width(t))
(t:SIntType) : SIntValue(v, width(t))
- expterm = (?op:#sym(?es:#exp ... ?ints:#int ... ?rest ...)) :
+ expterm = (?op:#sym(?es:#exp ... ?ints:#bigint ... ?rest ...)) :
if not empty?(rest) :
FPE(rest, "Illegal operands to primitive operator.")
match(primop(op)) :
- (p:PrimOp) : DoPrim(p, es, ints, UnknownType())
+ (p:PrimOp) : DoPrim(p, es, map(get{_,0},ints), UnknownType())
(p:False) : FPE(form, "Unrecognized primitive operator '~'." << [op])
expterm = (?op:#sym) :
Ref(op, UnknownType())
diff --git a/src/main/stanza/passes.stanza b/src/main/stanza/passes.stanza
index 33ef666c..255354a3 100644
--- a/src/main/stanza/passes.stanza
+++ b/src/main/stanza/passes.stanza
@@ -6,6 +6,7 @@ defpackage firrtl/passes :
import firrtl/primops
import firrtl-main
import firrtl/errors
+ import bigint
;============== Pass List ================
public val standard-passes = to-list $ [
@@ -1150,7 +1151,7 @@ defn expand-connect-indexed-stmt (s: Stmt,sh:HashTable<Symbol,Int>) -> Stmt :
to-list $
for (i in 0 to false, l in locs(s)) stream : Conditionally(
info(s),
- equality(ref,UIntValue(to-long $ i,UnknownWidth())),
+ equality(ref,UIntValue(BigIntLit(i),UnknownWidth())),
Connect(info(s),l,exp(s)),
EmptyStmt()
)
@@ -1164,7 +1165,7 @@ defn expand-connect-indexed-stmt (s: Stmt,sh:HashTable<Symbol,Int>) -> Stmt :
to-list $
for (i in 1 to false, e in tail(exps(s))) stream : Conditionally(
info(s),
- equality(ref,UIntValue(to-long $ i,UnknownWidth())),
+ equality(ref,UIntValue(BigIntLit(i),UnknownWidth())),
Connect(info(s),loc(s),e),
EmptyStmt()
)
@@ -1210,8 +1211,8 @@ public defmethod short-name (b:ExpandWhens) -> String : "expand-whens"
; ======== Expression Computation Library ===========
-val zero = UIntValue(to-long $ 0,IntWidth(1))
-val one = UIntValue(to-long $ 1,IntWidth(1))
+val zero = UIntValue(BigIntLit("b0"),IntWidth(1))
+val one = UIntValue(BigIntLit("b1"),IntWidth(1))
defmethod equal? (e1:Expression,e2:Expression) -> True|False :
match(e1,e2) :
@@ -1547,7 +1548,7 @@ public defn expand-whens (c:Circuit) -> Circuit :
(e:Expression) :
val ref = WRef(name(s),type(s),NodeKind(),FEMALE)
val en = to-exp(optimize $ get-write-enable(table[name(s)])) as Expression
- if en == UIntValue(to-long(1),UnknownWidth()) :
+ if en == UIntValue(BigIntLit("b1"),UnknownWidth()) :
add{cons,_} $ Connect(info(s),ref,e)
else :
add{cons,_} $ Conditionally(info(s),en,Connect(info(s),ref,e),EmptyStmt())
@@ -1562,7 +1563,7 @@ public defn expand-whens (c:Circuit) -> Circuit :
match(e) :
(e:Expression) :
val en = (to-exp $ optimize $ get-write-enable(table[n])) as Expression
- if en == UIntValue(to-long(1),UnknownWidth()) :
+ if en == UIntValue(BigIntLit("b1"),UnknownWidth()) :
add{cons,_} $ Connect(info(s),ref,e)
else :
add{cons,_} $ Conditionally(info(s),en,Connect(info(s),ref,e),EmptyStmt())
@@ -1957,14 +1958,14 @@ defn gen-constraints (m:Module, h:HashTable<Symbol,Type>, v:Vector<WGeq>) -> Mod
match(width(e)) :
(w:UnknownWidth) :
val w* = VarWidth(firrtl-gensym(`w,width-name-hash))
- add(v,WGeq(w*,IntWidth(to-int $ to-string $ ceil-log2(value(e)))))
+ add(v,WGeq(w*,IntWidth(length(to-bin(value(e))))))
UIntValue(value(e),w*)
(w) : e
(e:SIntValue) :
match(width(e)) :
(w:UnknownWidth) :
val w* = VarWidth(firrtl-gensym(`w,width-name-hash))
- add(v,WGeq(w*,IntWidth(1 + to-int $ to-string $ ceil-log2(abs(value(e))))))
+ add(v,WGeq(w*,IntWidth(1 + length(to-bin(value(e))))))
SIntValue(value(e),w*)
(w) : e
(e) : e
diff --git a/src/main/stanza/verilog.stanza b/src/main/stanza/verilog.stanza
index 6d9e2962..d39c89e4 100644
--- a/src/main/stanza/verilog.stanza
+++ b/src/main/stanza/verilog.stanza
@@ -65,8 +65,8 @@ defn emit-signed-if-any (e:Expression,ls:List<Expression>) -> String :
defn emit (e:Expression) -> String :
match(e) :
(e:Ref) : to-string $ name(e)
- (e:UIntValue) : string-join $ [width!(type(e)) "'d" value(e)]
- (e:SIntValue) : string-join $ [width!(type(e)) "'sd" value(e)]
+ (e:UIntValue) : string-join $ [width!(type(e)) "'" value(e)]
+ (e:SIntValue) : string-join $ [width!(type(e)) "'s" value(e)]
(e:Subfield) : error("Non-supported expression")
(e:Index) : error("Non-supported expression")
(e:DoPrim) :