Commit fd894614 authored by Hejing Li's avatar Hejing Li
Browse files

add no_simbricks attribute to exp class

parent 8b66ece4
...@@ -94,7 +94,7 @@ elif args.runtime == 'slurm': ...@@ -94,7 +94,7 @@ elif args.runtime == 'slurm':
else: else:
rt = runtime.LocalSimpleRuntime(verbose=args.verbose) rt = runtime.LocalSimpleRuntime(verbose=args.verbose)
def add_exp(e, run, prereq, create_cp, restore_cp): def add_exp(e, run, prereq, create_cp, restore_cp, no_simbricks):
outpath = '%s/%s-%d.json' % (args.outdir, e.name, run) outpath = '%s/%s-%d.json' % (args.outdir, e.name, run)
if os.path.exists(outpath) and not args.force: if os.path.exists(outpath) and not args.force:
print('skip %s run %d' % (e.name, run)) print('skip %s run %d' % (e.name, run))
...@@ -106,6 +106,7 @@ def add_exp(e, run, prereq, create_cp, restore_cp): ...@@ -106,6 +106,7 @@ def add_exp(e, run, prereq, create_cp, restore_cp):
env = exp.ExpEnv(args.repo, workdir, cpdir) env = exp.ExpEnv(args.repo, workdir, cpdir)
env.create_cp = create_cp env.create_cp = create_cp
env.restore_cp = restore_cp env.restore_cp = restore_cp
env.no_simbricks=no_simbricks
env.pcap_file = '' env.pcap_file = ''
if args.pcap: if args.pcap:
env.pcap_file = workdir+'/pcap' env.pcap_file = workdir+'/pcap'
...@@ -138,13 +139,17 @@ if not args.pickled: ...@@ -138,13 +139,17 @@ if not args.pickled:
continue continue
# if this is an experiment with a checkpoint we might have to create it # if this is an experiment with a checkpoint we might have to create it
if e.no_simbricks:
no_simbricks = True
else:
no_simbricks = False
if e.checkpoint: if e.checkpoint:
prereq = add_exp(e, 0, None, True, False) prereq = add_exp(e, 0, None, True, False, no_simbricks)
else: else:
prereq = None prereq = None
for run in range(args.firstrun, args.firstrun + args.runs): for run in range(args.firstrun, args.firstrun + args.runs):
add_exp(e, run, prereq, False, e.checkpoint) add_exp(e, run, prereq, False, e.checkpoint, no_simbricks)
else: else:
# otherwise load pickled run object # otherwise load pickled run object
for path in args.experiments: for path in args.experiments:
......
...@@ -32,6 +32,7 @@ class Experiment(object): ...@@ -32,6 +32,7 @@ class Experiment(object):
name = None name = None
timeout = None timeout = None
checkpoint = False checkpoint = False
no_simbricks = False
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
......
...@@ -197,6 +197,11 @@ class Gem5Host(HostSim): ...@@ -197,6 +197,11 @@ class Gem5Host(HostSim):
f'--num-cpus={self.node_config.cores} ' f'--num-cpus={self.node_config.cores} '
'--ddio-enabled --ddio-way-part=8 --mem-type=DDR4_2400_16x4 ') '--ddio-enabled --ddio-way-part=8 --mem-type=DDR4_2400_16x4 ')
if (env.no_simbricks):
cmd += '--no-simbricks '
if env.restore_cp: if env.restore_cp:
cmd += '-r 0 ' cmd += '-r 0 '
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment