returning working directory from Keras frontend

llvm/projects/keras/frontend/approxhpvm_translator.py

@@ -827,7 +827,8 @@ class TensorRtTranslator:
 
     self.add_header()
     
-    dir_path = "std::string dir_prefix = std::string(\"" + weights_dir + "\"); \n"
+    #dir_path = "std::string dir_prefix = std::string(\"" + weights_dir + "\"); \n"
+    dir_path = "std::string dir_prefix = std::string(\"../\"); \n"
     self.weight_str += dir_path
 
     if test_data is not None:
@@ -966,3 +967,5 @@ def translate_to_approxhpvm(model, weights_dir, test_data=None, test_labels=None
   hpvmTranslator.translate(model, weights_dir, test_data)
   
   
+  return weights_dir
+

llvm/projects/keras/src/Benchmark.py

@@ -60,8 +60,11 @@ class Benchmark:
           print("ERROR: Must load HPVM model to invoke frontend - use 'hpvm_reload'")
           sys.exit(1)
 
-        translate_to_approxhpvm(model, self.hpvm_dir, X_test, Y_test, self.num_classes)
+        working_dir = translate_to_approxhpvm(model, self.hpvm_dir, X_test, Y_test, self.num_classes)
+        print ("*** working_dir = ", working_dir)
+        
 
 
+