1 files changed, 7 insertions, 4 deletions
diff --git a/src/gpgpu-sim/mem_latency_stat.h b/src/gpgpu-sim/mem_latency_stat.h
index 982b9ae..0c84972 100644
--- a/src/gpgpu-sim/mem_latency_stat.h
+++ b/src/gpgpu-sim/mem_latency_stat.h
@@ -32,11 +32,13 @@
 #include <zlib.h>
 #include <map>
 
+class memory_config;
 class memory_stats_t {
 public:
    memory_stats_t( unsigned n_shader, 
-                   const struct shader_core_config *shader_config, 
-                   const struct memory_config *mem_config );
+                   const class shader_core_config *shader_config, 
+                   const memory_config *mem_config,
+				   const class gpgpu_sim* gpu);
 
    unsigned memlatstat_done( class mem_fetch *mf );
    void memlatstat_read_done( class mem_fetch *mf );
@@ -52,8 +54,9 @@ public:
 
    unsigned m_n_shader;
 
-   const struct shader_core_config *m_shader_config;
-   const struct memory_config *m_memory_config;
+   const shader_core_config *m_shader_config;
+   const memory_config *m_memory_config;
+   const class gpgpu_sim* m_gpu;
 
    unsigned max_mrq_latency;
    unsigned max_dq_latency;