Commit 8edc10eb authored by Alan Turner's avatar Alan Turner
Browse files

Set envvar

parent b56b11f1
import os, json, subprocess, tempfile, sys, argparse, contextlib, time
import os, subprocess, argparse, time
import tune_ck as tc
......@@ -42,6 +42,8 @@ def tune_models(models, batch_sizes, seq_len, n):
shell=True)
tc.tune(log_file, n, json_file)
tuning_path = os.path.abspath(json_file)
os.environ["MIGRAPHX_CK_TUNING"] = tuning_path
print("\nTuning results have been saved to:\n{}\n".format(json_file))
......
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