add spec unit params set

1 files changed, 7 insertions, 0 deletions

diff --git a/src/trace-driven/trace_driven.cc b/src/trace-driven/trace_driven.cc
index 0b1e24b..83b134e 100644
--- a/src/trace-driven/trace_driven.cc
+++ b/src/trace-driven/trace_driven.cc
@@ -665,6 +665,13 @@ void trace_config::set_latency(unsigned category, unsigned& latency,
     default:
       break;
   }
+  // for specialized units
+  if (category >= SPEC_UNIT_START_ID) {
+    unsigned spec_id = category - SPEC_UNIT_START_ID;
+    assert(spec_id >= 0 && spec_id < SPECIALIZED_UNIT_NUM);
+    latency = specialized_unit_latency[spec_id];
+    initiation_interval = specialized_unit_initiation[spec_id];
+  }
 }
 
 void trace_gpgpu_sim::createSIMTCluster() {