def _GetQueensEnvironmentArgs(n: int, delayed_reward: bool):
return {
'entry_point': _GetEntryPoint(delayed_reward),
'kwargs': {
'config': random_opt_pb2.Environment(
input_src=llvm_test_suite.SingleSource.Benchmarks.McGill.queens.srcs,
# Generate a gold standard using the binary. The assumes that the base
# build (before any opt passes have been run) is correct.
setup_cmd=f'$@ {n} > @D/gold_standard_output.txt',
exec_cmd=f'$@ {n} > @D/output.txt',
eval_cmd='cmp --silent @D/gold_standard_output.txt @D/output.txt',
candidate_pass=DEFAULT_PASS_LIST,
)
}
}
def _GetBzip2EnvironmentArgs(dataset_size: str, delayed_reward: bool):
return {
'entry_point': _GetEntryPoint(delayed_reward),
'kwargs': {
'config': random_opt_pb2.Environment(
input_src=bzip2.Bzip2.srcs,
# Create random data for bzip to compress.
setup_cmd=f'head -c {dataset_size} </dev/urandom > @D/input.dat',
# Compress and deflate the input data.
exec_cmd=('$@ -z < @D/input.dat > @D/input.dat.bz2 && '
'$@ -d < @D/input.dat.bz2 > @D/output.dat'),
eval_cmd='cmp --silent @D/input.dat @D/output.dat',
candidate_pass=DEFAULT_PASS_LIST,
)
}
}
def _GetBzip2EnvironmentArgs(dataset_size: str, delayed_reward: bool):
return {
"entry_point": _GetEntryPoint(delayed_reward),
"kwargs": {
"config":
random_opt_pb2.Environment(
input_src=bzip2.Bzip2.srcs,
# Create random data for bzip to compress.
setup_cmd=
f"head -c {dataset_size} </dev/urandom > @D/input.dat",
# Compress and deflate the input data.
exec_cmd=("$@ -z < @D/input.dat > @D/input.dat.bz2 && "
"$@ -d < @D/input.dat.bz2 > @D/output.dat"),
eval_cmd="cmp --silent @D/input.dat @D/output.dat",
candidate_pass=DEFAULT_PASS_LIST,
)
},
}