From d2c758555b138fc7baa8b12e3c64a27988d5b6ae Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 11 Mar 2021 00:33:36 -0600
Subject: [PATCH] Adding automatic HPVM compile option to Keras frontend
---
.../keras/frontend/approxhpvm_translator.py | 2 +-
hpvm/projects/keras/src/Benchmark.py | 39 ++++++++++++++-----
2 files changed, 30 insertions(+), 11 deletions(-)
diff --git a/hpvm/projects/keras/frontend/approxhpvm_translator.py b/hpvm/projects/keras/frontend/approxhpvm_translator.py
index 1b041a6de4..852b3e89f4 100644
--- a/hpvm/projects/keras/frontend/approxhpvm_translator.py
+++ b/hpvm/projects/keras/frontend/approxhpvm_translator.py
@@ -1127,5 +1127,5 @@ def translate_to_approxhpvm(model,
print ("-- ApproxHPVM src : ", src_dir + "approxhpvm_src.cc")
- return weights_dir
+ return src_dir
diff --git a/hpvm/projects/keras/src/Benchmark.py b/hpvm/projects/keras/src/Benchmark.py
index 7b0b5447bc..b3b85a4277 100644
--- a/hpvm/projects/keras/src/Benchmark.py
+++ b/hpvm/projects/keras/src/Benchmark.py
@@ -3,6 +3,7 @@
import sys
import os
import shutil
+import subprocess
from keras.utils.np_utils import to_categorical
from keras.models import load_model
from frontend.approxhpvm_translator import translate_to_approxhpvm
@@ -39,18 +40,36 @@ class Benchmark:
# Compiles frontend generated sources
def compileSource(self, working_dir):
+
+ src_file = os.getcwd() + "/" + working_dir + "/approxhpvm_src.cc"
+ target_binary = os.getcwd() + "/" + working_dir + "/HPVM_binary"
+ approx_conf_file = "tuner_confs.txt"
- # set LLVM_SRC_ROOT
- os.environ["CFLAGS"] = ""
- os.environ["CXXFLAGS"] = ""
-
- dest_file = working_dir + "CMakeLists.txt"
- shutil.copy("cmake_template/CMakeLists.txt", dest_file)
-
- # Cmake ../
- # make
-
+ FNULL = open(os.devnull, 'w')
+
+ try:
+ subprocess.run([
+ "approxhpvm.py",
+ "-h"
+ ], check=True, stdout=FNULL)
+
+ except:
+ print ("\n\n ERROR: Could not find approxhpvm.py (HPVM compile script)!! \n\n")
+ print ("To Compile, Must set PATH to include approxhpvm.py script. Do the following: ")
+ print ("**** export PATH=${PATH_TO_YOUR_HPVM_INSTALLATION}/build/bin/:$PATH *****")
+ sys.exit(1)
+
+
+ try:
+ subprocess.run([
+ "approxhpvm.py", src_file, target_binary,
+ "-t", "tensor", "--conf-file", approx_conf_file
+ ], check=True)
+ except:
+ print ("\n\n ERROR: HPVM Compilation Failed!! \n\n")
+ sys.exit(1)
+
def printUsage(self):
print ("Usage: python ${benchmark.py} [hpvm_reload|train] [frontend] [compile]")
--
GitLab