Set up proper Config system for numLanes

TODO: tracefilename should not really be inside SIMTCoreParam.
This commit is contained in:
Hansung Kim
2023-05-08 17:32:31 -07:00
parent 25c0b6cfa5
commit f6be54a122
2 changed files with 25 additions and 38 deletions

View File

@@ -1,44 +1,25 @@
package freechips.rocketchip.tilelink
import chisel3._
import chisel3.util._
import freechips.rocketchip.diplomacy._
import freechips.rocketchip.subsystem.{BaseSubsystem, CacheBlockBytes}
import org.chipsalliance.cde.config.{Parameters, Field, Config}
import freechips.rocketchip.subsystem.{BaseSubsystem}
import org.chipsalliance.cde.config.{Parameters, Config}
// class class, consumed by WithGPUTacer config and GPUTracerKey
case class GPUTracerConfig(numLanes: Int, traceFile : String) // FIXME, add lane number and file name
case object GPUTracerKey extends Field[Option[GPUTracerConfig]](None)
// Both LazyModule of Tracer and Impl are both in Coalescing.scala
//The trait is attached to DigitalTop of Chipyard system, informing it indeed has the ability
//to attach GPU tracer node onto the system bus
// The trait is attached to DigitalTop of Chipyard system, informing it indeed
// has the ability to attach GPU tracer node onto the system bus
trait CanHaveGPUTracer { this: BaseSubsystem =>
implicit val p: Parameters
//p(GPUTracerKey) is the mechnimism to pass Config's parameter down to lazymodule
p(GPUTracerKey) .map { k =>
val config = p(GPUTracerKey).get
val tracer = LazyModule(new MemTraceDriver(defaultConfig, config.traceFile)(p))
// Must use :=* to ensure the N edges from Tracer doesn't get merged into 1 when connecting to SBus
p(SIMTCoreKey).map { _ =>
val config = p(SIMTCoreKey).get
val tracer = LazyModule(new MemTraceDriver(defaultConfig, config.tracefilename)(p))
// Must use :=* to ensure the N edges from Tracer doesn't get merged into 1
// when connecting to SBus
sbus.fromPort(Some("gpu-tracer"))() :=* tracer.node
}
}
//This is used by Chip Level Config, the config which creates the SoC
class WithGPUTracer(numLanes: Int, traceFile : String) extends Config((site, here, up) => {
case GPUTracerKey => Some( GPUTracerConfig(numLanes, traceFile) )
}
)
class WithGPUTracer(numLanes: Int, tracefilename: String)
extends Config((_, _, _) => { case SIMTCoreKey =>
Some(SIMTCoreParams(numLanes, tracefilename))
})