Chisel:从Tester访问模块参数

时间:2015-05-20 22:54:44

标签: scala chisel

如何从正在测试它的测试程序内部访问用于构造模块的参数?

在下面的测试中,我将参数显式传递给Module和Tester。我宁愿不必将它们传递给Tester,而是将它们从传入的模块中提取出来。

我也是scala / chisel的新手,所以我对使用不良技术的任何提示都会很感激:)。

import Chisel._
import math.pow

class TestA(dataWidth: Int, arrayLength: Int) extends Module {
  val dataType = Bits(INPUT, width = dataWidth)
  val arrayType = Vec(gen = dataType, n = arrayLength)
  val io = new Bundle {
    val i_valid = Bool(INPUT)
    val i_data = dataType
    val i_array = arrayType
    val o_valid = Bool(OUTPUT)
    val o_data = dataType.flip
    val o_array = arrayType.flip
  }
  io.o_valid := io.i_valid
  io.o_data := io.i_data
  io.o_array := io.i_array
}

class TestATests(c: TestA, dataWidth: Int, arrayLength: Int) extends Tester(c) {
  val maxData = pow(2, dataWidth).toInt
  for (t <- 0 until 16) {
    val i_valid = rnd.nextInt(2)
    val i_data = rnd.nextInt(maxData)
    val i_array = List.fill(arrayLength)(rnd.nextInt(maxData))
    poke(c.io.i_valid, i_valid)
    poke(c.io.i_data, i_data)
    (c.io.i_array, i_array).zipped foreach {
      (element,value) => poke(element, value)
    }
    expect(c.io.o_valid, i_valid)
    expect(c.io.o_data, i_data)
    (c.io.o_array, i_array).zipped foreach {
      (element,value) => poke(element, value)
    }
    step(1)
  }
}    

object TestAObject {
  def main(args: Array[String]): Unit = {
    val tutArgs = args.slice(0, args.length)
    val dataWidth = 5
    val arrayLength = 6
    chiselMainTest(tutArgs, () => Module(
      new TestA(dataWidth=dataWidth, arrayLength=arrayLength))){
      c => new TestATests(c, dataWidth=dataWidth, arrayLength=arrayLength)
    }
   }
}

1 个答案:

答案 0 :(得分:2)

如果您创建TestA的参数dataWidth和arrayLength成员,您只需引用它们即可。在Scala中,这可以通过在参数列表中插入val来实现:

class TestA(val dataWidth: Int, val arrayLength: Int) extends Module ...

然后,您可以使用c.dataWidthc.arrayLength

作为成员从测试中引用它们