Num. CPUs produced by trace_reader, not hardcoded

1 files changed, 8 insertions, 0 deletions

diff --git a/trace_reader.py b/trace_reader.py
index 31f1812..2e5f65d 100644
--- a/trace_reader.py
+++ b/trace_reader.py
@@ -35,6 +35,14 @@ import struct
 # Generator function returning an iterable over records in a trace file.
 def trace_reader(files):
 
+    # Yield a record indicating the number of CPUs, used by the G-EDF test
+    class Obj: pass
+    record = Obj()
+    record.record_type = "meta"
+    record.type_name = "num_cpus"
+    record.num_cpus = len(files)
+    yield record
+
     # Create iterators for each file and a buffer to store records in
     file_iters = [] # file iterators
     file_iter_buff = [] # file iterator buffers