Commit edd3e20b authored by Tomáš Oberhuber's avatar Tomáš Oberhuber
Browse files

Fixing Python script for processing of SpMV benchmark results.

parent e5170512

src/Benchmarks/scripts/tnl-spmv-benchmark-make-tables-json.py

+24 −12
Original line number Diff line number Diff line
@@ -632,24 +632,28 @@ def cusparse_speedup_comparison( df, formats, head_size=10 ): 
   for format in formats:

      if( 'Ellpack' in format and not 'Symmetric' in format and not 'Binary' in format and not 'Legacy' in format ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "ellpack-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )



   current_formats.clear()

   for format in formats:

      if( 'Ellpack' in format and 'Symmetric' in format and not 'Binary' in format and not 'Legacy' in format ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "symmetric-ellpack-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )



   current_formats.clear()

   for format in formats:

      if( 'Ellpack' in format and not 'Symmetric' in format and 'Binary' in format and not 'Legacy' in format ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "binary-ellpack-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )



   current_formats.clear()

   for format in formats:

      if( 'Ellpack' in format and 'Symmetric' in format and 'Binary' in format and not 'Legacy' in format ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "symmetric-binary-ellpack-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )
@@ -658,23 +662,27 @@ def cusparse_speedup_comparison( df, formats, head_size=10 ): 
   for format in formats:

      if( 'CSR' in format and not 'Symmetric' in format and not 'Binary' in format and not 'Legacy' in format and not 'Hybrid' in format and format != 'CSR' ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "csr-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )

   current_formats.clear()

   for format in formats:

      if( 'CSR' in format and 'Symmetric' in format and not 'Binary' in format and not 'Legacy' in format and not 'Hybrid' in format and format != 'CSR' ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "symmetric-csr-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )

   current_formats.clear()



   for format in formats:

      if( 'CSR' in format and not 'Symmetric' in format and 'Binary' in format and not 'Legacy' in format and not 'Hybrid' in format and format != 'CSR' ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "binary-csr-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )

   current_formats.clear()



   for format in formats:

      if( 'CSR' in format and 'Symmetric' in format and 'Binary' in format and not 'Legacy' in format and not 'Hybrid' in format and format != 'CSR' ):

         current_formats.append( format )

   if current_formats:

      draw_profiles( current_formats, profiles, xlabel, ylabel, "-symmetric-binary-csr-profiles-cusparse-speedup.pdf", 'upper right', "cuSPARSE" )

   current_formats.clear()
@@ -1111,9 +1119,13 @@ input_df.to_html( "sparse-matrix-benchmark-test.html") 




formats = list(set( input_df['format'].values.tolist() )) # list of all formats in the benchmark results

if 'CSR< Light > Automatic' in formats:

   formats.remove('CSR< Light > Automatic')

if 'Binary CSR< Light > Automatic' in formats:

   formats.remove('Binary CSR< Light > Automatic')

if 'Symmetric CSR< Light > Automatic' in formats:

   formats.remove('Symmetric CSR< Light > Automatic')

if 'Symmetric Binary CSR< Light > Automatic' in formats:

   formats.remove('Symmetric Binary CSR< Light > Automatic')

formats.append('TNL Best')

formats.append('CSR Light Best')