Warmup optional (#18)

* optional warmup flag

---------

Co-authored-by: Vamsi Subraveti <vis@ornl.gov>
Co-authored-by: John Coleman <103219962+colemanjs@users.noreply.github.com>

Author: 3 people

examples/api_example/rve.py

@@ -76,15 +76,9 @@
 }
 
 melt_pool = create_melt_pool(melt_pool_dict, enable_random_phases=True)
-for path_vector in path_vectors:
-    path_vector.set_melt_pool_properties(melt_pool)
 
 # 4. Compute porosity using conic section / superellipse curves for melt pool mask
-porosity = compute_porosity(
-    grid,
-    path_vectors,
-    melt_pool,
-)
+porosity = compute_porosity(grid, path_vectors, melt_pool, jit_warmup=True)
 
 # 5. Write porosity field to .VTI
 write_vtk(grid.origin, grid.resolution, porosity, "rve.vti")

src/raptor/api.py

@@ -155,24 +155,28 @@ def create_melt_pool(
 
 
 def compute_porosity(
-    grid: Grid, path_vectors: List[PathVector], melt_pool: MeltPool
+    grid: Grid,
+    path_vectors: List[PathVector],
+    melt_pool: MeltPool,
+    jit_warmup: Optional[bool] = True,
 ) -> None:
     """
     Main computation: computes porosity field.
     """
 
-    print("JIT Warmup...")
-    t_start_warmup = time.time()
+    if jit_warmup:
+        print("JIT Warmup...")
+        t_start_warmup = time.time()
 
-    # Warm up the vector property assignment.
-    if path_vectors:
-        path_vectors[0].set_melt_pool_properties(melt_pool)
+        # Warm up the vector property assignment.
+        if path_vectors:
+            path_vectors[0].set_melt_pool_properties(melt_pool)
 
-    # Warm up the main, parallelized compute kernel.
-    if grid.n_voxels > 0 and path_vectors:
-        _ = compute_melt_mask(grid.voxels[0:1], melt_pool, path_vectors[0:1])
+        # Warm up the main, parallelized compute kernel.
+        if grid.n_voxels > 0 and path_vectors:
+            _ = compute_melt_mask(grid.voxels[0:1], melt_pool, path_vectors[0:1])
 
-    print(f" -> JIT warmup complete ({time.time() - t_start_warmup:.8f}s).")
+        print(f" -> JIT warmup complete ({time.time() - t_start_warmup:.8f}s).")
 
     print(f"Preparing {len(path_vectors)} path vectors for simulation...")
     t0_setup = time.time()