Adding automatic HPVM compile option to Keras frontend

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
 

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]")