Fixing issue with running outside of examples dir

Author: aolofsson

examples/baremetal/make.py

@@ -28,18 +28,21 @@
 
     parser.add_argument("-group","-g",
                         nargs='+',
-                        choices=['basic', 'memory', 'arithmetic', 'epfl', 'block'],
+                        choices=['basic',
+                                 'memory',
+                                 'arithmetic',
+                                 'epfl',
+                                 'block'],
                         required=True,
                         help="Benchmark group")
     parser.add_argument("-name","-n",
                         nargs='+',
                         help="Benchmark name")
     parser.add_argument("-tool",
                         choices=['yosys', 'vivado'],
-                        required=True,
+                        default="yosys",
                         help="Tool name")
     parser.add_argument("-target",
-                        required=True,
                         help="Compilation target")
     parser.add_argument('-clean','-c',
                         action='store_true',
@@ -51,11 +54,12 @@
     args = parser.parse_args()
 
     # resolving relative path
+    cwd = os.getcwd()
     scriptdir = Path(__file__).resolve().parent
     rootdir = Path(__file__).resolve().parent.parent.parent
 
     # generated local script
-    env = Environment(loader=FileSystemLoader('.'))
+    env = Environment(loader=FileSystemLoader(scriptdir))
     template = env.get_template(f'{args.tool}_template.j2')
 
     # global analysis
@@ -79,11 +83,12 @@
                 script = f"{name}.tcl"
                 cmd = ['vivado', '-mode batch', '-source', script]
 
-            # clean up old run
-            if args.clean: # create run dir.clean:
-                shutil.rmtree(f"build/{group}/{name}")
+            # clean up old results
+            if os.path.isdir(f"build/{group}/{name}"):
+                if args.clean: # create run dir.clean:
+                    shutil.rmtree(f"build/{group}/{name}")
 
-            # create run dir
+            # change dir
             os.makedirs(f"build/{group}/{name}", exist_ok=True)
             os.chdir(f"build/{group}/{name}")
 
@@ -131,8 +136,8 @@
                     data = json.load(f)
                 results["cells"][name] = data["design"]["num_cells"]
 
-            # go back home
-            os.chdir(scriptdir)
+            # go back to cwd
+            os.chdir(cwd)
 
 
     # writing results to file