jax-hpc-profiler 0.2.7__tar.gz → 0.2.8__tar.gz

{jax_hpc_profiler-0.2.7 → jax_hpc_profiler-0.2.8}/PKG-INFO

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

{jax_hpc_profiler-0.2.7 → jax_hpc_profiler-0.2.8}/pyproject.toml

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

{jax_hpc_profiler-0.2.7 → jax_hpc_profiler-0.2.8}/src/jax_hpc_profiler/plotting.py

@@ -7,7 +7,7 @@ import pandas as pd
 from matplotlib.axes import Axes
 from matplotlib.patches import FancyBboxPatch
 
-from .utils import inspect_df, plot_with_pdims_strategy
+from .utils import inspect_df, plot_with_pdims_strategy, inspect_data
 
 np.seterr(divide='ignore')
 plt.rcParams.update({'font.size': 15})
@@ -118,13 +118,13 @@ def plot_scaling(dataframes: Dict[str, pd.DataFrame],
         axs = axs.flatten()
     else:
         axs = [axs]
-
+    
     for i, fixed_size in enumerate(fixed_sizes):
         ax: Axes = axs[i]
 
+        x_values = []
+        y_values = []
         for method, df in dataframes.items():
-            x_values = []
-            y_values = []
 
             filtered_method_df = df[df[fixed_column] == int(fixed_size)]
             if filtered_method_df.empty:
@@ -137,6 +137,7 @@ def plot_scaling(dataframes: Dict[str, pd.DataFrame],
                                    plot_columns)
 
             for backend, precision, function, plot_column in combinations:
+              
                 filtered_params_df = filtered_method_df[
                     (filtered_method_df['backend'] == backend)
                     & (filtered_method_df['precision'] == precision) &
@@ -149,10 +150,11 @@ def plot_scaling(dataframes: Dict[str, pd.DataFrame],
 
                 x_values.extend(x_vals)
                 y_values.extend(y_vals)
-
-        plotting_memory = 'time' not in plot_columns[0].lower()
-        configure_axes(ax, x_values, y_values, f"{title} {fixed_size}", xlabel,
-                       plotting_memory, memory_units)
+        
+        if len(x_values) != 0: 
+          plotting_memory = 'time' not in plot_columns[0].lower()
+          configure_axes(ax, x_values, y_values, f"{title} {fixed_size}", xlabel,
+                         plotting_memory, memory_units)
 
     for i in range(num_subplots, num_rows * num_cols):
         fig.delaxes(axs[i])

{jax_hpc_profiler-0.2.7 → jax_hpc_profiler-0.2.8}/src/jax_hpc_profiler/utils.py

@@ -336,12 +336,11 @@ def clean_up_csv(
         # Filter data sizes
         if data_sizes:
             df = df[df['x'].isin(data_sizes)]
-
+        
         # Filter pdims
         if pdims:
             px_list, py_list = zip(*[map(int, p.split('x')) for p in pdims])
             df = df[(df['px'].isin(px_list)) & (df['py'].isin(py_list))]
-
         # convert memory units columns to remquested memory_units
         match memory_units:
             case 'KB':
@@ -385,6 +384,7 @@ def clean_up_csv(
             df.drop(columns=['px', 'py'], inplace=True)
             if not 'plot_all' in pdims_strategy:
                 df = df[df['decomp'].isin(pdims_strategy)]
+
         # check available gpus in dataset
         available_gpu_counts.update(df['gpus'].unique())
         available_data_sizes.update(df['x'].unique())

{jax_hpc_profiler-0.2.7 → jax_hpc_profiler-0.2.8}/src/jax_hpc_profiler.egg-info/PKG-INFO

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