From 789d5da12a6c719dbfc94b4e1674321dd1455884 Mon Sep 17 00:00:00 2001 From: Yifan Zhao <yifanz16@illinois.edu> Date: Mon, 25 Jan 2021 03:14:05 -0600 Subject: [PATCH] Added env file and readme --- README.md | 16 ++++++++++++++++ env.yaml | 19 +++++++++++++++++++ 2 files changed, 35 insertions(+) create mode 100644 README.md create mode 100644 env.yaml diff --git a/README.md b/README.md new file mode 100644 index 0000000..41c4809 --- /dev/null +++ b/README.md @@ -0,0 +1,16 @@ +# Autotuning and Predictive Autotuning + +Performs autotuning on program approximation knobs using an error-predictive proxy in place of the +original program, to greatly speedup autotuning while getting results comparable in quality. + +Work in progress. + +## Requirements + +Prerequisite packages are listed in `./env.yaml`. Conda is the validated and recommended way to set +up a working environment. If you're using conda, do + +```bash +conda env create -n predtuner -f env.yaml +conda activate predtuner +``` diff --git a/env.yaml b/env.yaml new file mode 100644 index 0000000..735790b --- /dev/null +++ b/env.yaml @@ -0,0 +1,19 @@ +name: predtuner +channels: + - pytorch + - defaults +dependencies: + - matplotlib=3.3.2 + - networkx=2.5 + - python=3.8.5 + - pytorch=1.7.0 + - torchvision=0.8.1 + - tqdm=4.50.2 + - pandas=1.1.3 + - pip=20.2.4 + - wheel=0.35.1 + - jsonpickle=1.5 + - pip: + - argparse + - opentuner==0.8.3 # Must be 0.8.3, they fixed an important bug + - sqlalchemy==1.3 -- GitLab