From 5b35e8b6622ede8de4734a455aea818dc3c062f2 Mon Sep 17 00:00:00 2001
From: Yaqi Zhang <hsbzyq@gmail.com>
Date: Thu, 17 Jan 2019 11:52:25 -0800
Subject: [PATCH] add option to set pcu and pmu stages

---
 src/core/SpadeConfig.scala        | 2 ++
 src/param/routable/CUParams.scala | 4 ++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/core/SpadeConfig.scala b/src/core/SpadeConfig.scala
index 026f29b..e7775fc 100644
--- a/src/core/SpadeConfig.scala
+++ b/src/core/SpadeConfig.scala
@@ -43,5 +43,7 @@ object SpadeConfig extends prism.GlobalConfig {
   register[Int]("vc", default=4, info="Number of virtual classes per network")
   register[String]("link-prop", default="db", info="[db-double buffered, cd-credit based]")
   register[Int]("flit-width", default=512, info="Flit width for dynamic network")
+  register[Int]("pcu-stage", default=6, info="Number of stages in PCU")
+  register[Int]("pmu-stage", default=4, info="Number of stages in PMU")
 
 }
diff --git a/src/param/routable/CUParams.scala b/src/param/routable/CUParams.scala
index e38d9ba..84c4c30 100644
--- a/src/param/routable/CUParams.scala
+++ b/src/param/routable/CUParams.scala
@@ -31,7 +31,7 @@ case class PCUParam (
   vectorFifoParam:FIFOParam=FIFOParam(size=option("fifo-depth")),
   numCtrs:Int=6,
   simdParam:Option[SIMDParam]=Some(DefaultSIMDParam(
-    numStages=6, 
+    numStages=option[Int]("pcu-stage"), 
     ops=allOps, 
     vectorized=true, 
     numRegs=16, 
@@ -72,7 +72,7 @@ case class PMUParam (
   sramParam:SRAMParam=SRAMParam(size=option("pmu-sram-size"),4), // 256 kB
   numCtrs:Int=6,
   simdParam:Option[SIMDParam]=Some(DefaultSIMDParam(
-    numStages=4, 
+    numStages=option[Int]("pmu-stage"), 
     ops=fixOps ++ bitOps ++ otherOps, 
     vectorized=true, 
     numRegs=16,