jax-hpc-profiler 0.2.5__tar.gz → 0.2.6__tar.gz

{jax_hpc_profiler-0.2.5 → jax_hpc_profiler-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jax_hpc_profiler
-Version: 0.2.5
+Version: 0.2.6
 Summary: HPC Plotter and profiler for benchmarking data made for JAX
 Author: Wassim Kabalan
 License:                     GNU GENERAL PUBLIC LICENSE

{jax_hpc_profiler-0.2.5 → jax_hpc_profiler-0.2.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "jax_hpc_profiler"
-version = "0.2.5"
+version = "0.2.6"
 description = "HPC Plotter and profiler for benchmarking data made for JAX"
 authors = [
     { name="Wassim Kabalan" }

{jax_hpc_profiler-0.2.5 → jax_hpc_profiler-0.2.6}/src/jax_hpc_profiler/create_argparse.py

@@ -136,23 +136,25 @@ def create_argparser():
         default="%m%-%f%-%pn%-%pr%-%b%-%p%-%n%")
 
     args = parser.parse_args()
-
-    if 'plot_all' in args.pdim_strategy and len(args.pdim_strategy) > 1:
-        print(
-            "Warning: 'plot_all' strategy is combined with other strategies. Using 'plot_all' only."
-        )
-        args.pdim_strategy = ['plot_all']
-
-    if 'plot_fastest' in args.pdim_strategy and len(args.pdim_strategy) > 1:
-        print(
-            "Warning: 'plot_fastest' strategy is combined with other strategies. Using 'plot_fastest' only."
-        )
-        args.pdim_strategy = ['plot_fastest']
-    if args.plot_times is not None:
-        args.plot_columns = args.plot_times
-    elif args.plot_memory is not None:
-        args.plot_columns = args.plot_memory
-    else:
-        raise ValueError('Either plot_times or plot_memory should be provided')
+    
+    # if command was plot, then check if pdim_strategy is validat
+    if args.command == 'plot':
+      if 'plot_all' in args.pdim_strategy and len(args.pdim_strategy) > 1:
+          print(
+              "Warning: 'plot_all' strategy is combined with other strategies. Using 'plot_all' only."
+          )
+          args.pdim_strategy = ['plot_all']
+
+      if 'plot_fastest' in args.pdim_strategy and len(args.pdim_strategy) > 1:
+          print(
+              "Warning: 'plot_fastest' strategy is combined with other strategies. Using 'plot_fastest' only."
+          )
+          args.pdim_strategy = ['plot_fastest']
+      if args.plot_times is not None:
+          args.plot_columns = args.plot_times
+      elif args.plot_memory is not None:
+          args.plot_columns = args.plot_memory
+      else:
+          raise ValueError('Either plot_times or plot_memory should be provided')
 
     return args

{jax_hpc_profiler-0.2.5 → jax_hpc_profiler-0.2.6}/src/jax_hpc_profiler/timer.py

@@ -138,6 +138,7 @@ class Timer:
 
         times_array = self._get_mean_times()
         if jax.process_index() == 0:
+
             min_time = np.min(times_array)
             max_time = np.max(times_array)
             mean_time = np.mean(times_array)

{jax_hpc_profiler-0.2.5 → jax_hpc_profiler-0.2.6}/src/jax_hpc_profiler.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jax_hpc_profiler
-Version: 0.2.5
+Version: 0.2.6
 Summary: HPC Plotter and profiler for benchmarking data made for JAX
 Author: Wassim Kabalan
 License:                     GNU GENERAL PUBLIC LICENSE