wu-arch/radiance
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Do proper recoding and boxing for FMA input

Browse Source
This commit is contained in:
Hansung Kim
2024-05-28 16:41:44 -07:00
parent 615815acf5
commit 4dba0def01
2 changed files with 74 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/test/scala/radiance/TensorDPUTest.scala
View File

@@ -5,13 +5,16 @@ import chisel3.stage.PrintFullStackTraceAnnotation
import chisel3.util._
import chiseltest._
import chiseltest.simulator.VerilatorFlags
import org.chipsalliance.cde.config.Parameters
import freechips.rocketchip.tile
import org.scalatest.flatspec.AnyFlatSpec
class MulAddTest extends AnyFlatSpec with ChiselScalatestTester {
behavior of "MulAddRecFNPipe"
val t = tile.FType.S
it should "do basic arithmetic" in {
test(new MulAddRecFNPipe(2, 8, 23))
test(new MulAddRecFNPipe(2, t.exp, t.sig))
// .withAnnotations(Seq(WriteVcdAnnotation))
{ c =>
c.io.validin.poke(true.B)
@@ -24,18 +27,48 @@ class MulAddTest extends AnyFlatSpec with ChiselScalatestTester {
// 0: round to nearest, ties to even
c.io.roundingMode.poke(0.U)
c.io.detectTininess.poke(hardfloat.consts.tininess_beforeRounding)
c.io.a.poke(0x3f800000.U/*2.0*/)
c.io.b.poke(0x3f800000.U/*3.0*/)
c.io.c.poke(0x00000000.U/*0.0*/)
c.io.a.poke(0x3f800000.U)
c.io.b.poke(0x3f800000.U)
c.io.c.poke(0x00000000.U)
c.clock.step()
c.io.validin.poke(false.B)
c.io.validout.expect(false.B)
c.clock.step()
c.io.validout.expect(true.B)
c.io.out.expect(0x40c00000.U/*6.0*/)
c.io.out.expect(0x40c00000.U)
c.clock.step()
c.io.validout.expect(false.B)
}
}
}
class DPUPipeTest extends AnyFlatSpec with ChiselScalatestTester {
behavior of "DPUPipe"
implicit val p: Parameters = Parameters.empty
it should "pass" in {
test(new DPUPipe)
// .withAnnotations(Seq(WriteVcdAnnotation))
{ fma =>
fma.io.in.valid.poke(true.B)
fma.io.in.bits.a.poke(0x40000000L.U(64.W))
fma.io.in.bits.b.poke(0x40400000L.U(64.W))
fma.io.in.bits.c.poke(0x3f800000L.U(64.W))
fma.clock.step()
fma.io.in.valid.poke(true.B)
fma.io.in.bits.a.poke(0x40000000L.U(64.W))
fma.io.in.bits.b.poke(0x3f800000L.U(64.W))
fma.io.in.bits.c.poke(0x3f800000L.U(64.W))
fma.clock.step()
fma.io.in.valid.poke(false.B)
fma.io.out.valid.expect(true.B)
fma.io.out.bits.data.expect(0x40e00000L.U)
fma.clock.step()
// pipelined back-to-back response
fma.io.out.valid.expect(true.B)
fma.io.out.bits.data.expect(0x40400000L.U)
}
}
}