From 283c71e69a85e56285c078bea7c439f72a18bde0 Mon Sep 17 00:00:00 2001
From: Mahmoud <abdallm@purdue.edu>
Date: Sun, 31 May 2020 09:22:03 -0400
Subject: add spec unit params set

---
 src/trace-driven/trace_driven.cc | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/trace-driven/trace_driven.cc')

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() {
-- 
cgit v1.3