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 import tune_ck as tc
...@@ -42,6 +42,8 @@ def tune_models(models, batch_sizes, seq_len, n): ...@@ -42,6 +42,8 @@ def tune_models(models, batch_sizes, seq_len, n):
shell=True) shell=True)
tc.tune(log_file, n, json_file) 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)) 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