Fixed Python SpMV benchmark script. (1e1a974d) · Commits · TNL / tnl-dev

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

+57 −57

Original line number	Diff line number	Diff line
		@@ -214,19 +214,19 @@ for x in cuSparse_list:
		cuSparse_file.write( f"{i+1} {x} " ) # 1 2
		cuSparse_file.write( f"{cuSparse_csr_legacy_adaptive_gpu_list[ i ]} " ) # 3
		cuSparse_file.write( f"{cuSparse_csr_legacy_light_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light2_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light3_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light4_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light5_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light6_gpu_list[ i ]} " ) # 4
		cuSparse_file.write( f"{cuSparse_csr_legacy_light_without_atomic_gpu_list[ i ]} " ) # 5
		cuSparse_file.write( f"{cuSparse_csr_legacy_scalar_gpu_list[ i ]} " ) # 6
		cuSparse_file.write( f"{cuSparse_csr_legacy_vector_gpu_list[ i ]} " ) # 7
		cuSparse_file.write( f"{cuSparse_csr_legacy_multivector_gpu_list[ i ]} " ) # 8
		cuSparse_file.write( f"{cuSparse_ellpack_gpu_list[ i ]} {cuSparse_ellpack_legacy_gpu_list[ i ]} " ) # 9 10
		cuSparse_file.write( f"{cuSparse_sliced_ellpack_gpu_list[ i ]} {cuSparse_sliced_ellpack_legacy_gpu_list[ i ]} " ) # 11 12
		cuSparse_file.write( f"{cuSparse_chunked_ellpack_gpu_list[ i ]} {cuSparse_chunked_ellpack_legacy_gpu_list[ i ]} " ) # 13 14
		cuSparse_file.write( f"{cuSparse_bi_ellpack_gpu_list[ i ]} {cuSparse_bi_ellpack_legacy_gpu_list[ i ]}\n" ) # 15 16
		cuSparse_file.write( f"{cuSparse_csr_legacy_light2_gpu_list[ i ]} " ) # 5
		cuSparse_file.write( f"{cuSparse_csr_legacy_light3_gpu_list[ i ]} " ) # 6
		cuSparse_file.write( f"{cuSparse_csr_legacy_light4_gpu_list[ i ]} " ) # 7
		cuSparse_file.write( f"{cuSparse_csr_legacy_light5_gpu_list[ i ]} " ) # 8
		cuSparse_file.write( f"{cuSparse_csr_legacy_light6_gpu_list[ i ]} " ) # 9
		cuSparse_file.write( f"{cuSparse_csr_legacy_light_without_atomic_gpu_list[ i ]} " ) # 10
		cuSparse_file.write( f"{cuSparse_csr_legacy_scalar_gpu_list[ i ]} " ) # 11
		cuSparse_file.write( f"{cuSparse_csr_legacy_vector_gpu_list[ i ]} " ) # 12
		cuSparse_file.write( f"{cuSparse_csr_legacy_multivector_gpu_list[ i ]} " ) # 13
		cuSparse_file.write( f"{cuSparse_ellpack_gpu_list[ i ]} {cuSparse_ellpack_legacy_gpu_list[ i ]} " ) # 14 15
		cuSparse_file.write( f"{cuSparse_sliced_ellpack_gpu_list[ i ]} {cuSparse_sliced_ellpack_legacy_gpu_list[ i ]} " ) # 16 17
		cuSparse_file.write( f"{cuSparse_chunked_ellpack_gpu_list[ i ]} {cuSparse_chunked_ellpack_legacy_gpu_list[ i ]} " ) # 18 19
		cuSparse_file.write( f"{cuSparse_bi_ellpack_gpu_list[ i ]} {cuSparse_bi_ellpack_legacy_gpu_list[ i ]}\n" ) # 20 21
		i = i + 1
		cuSparse_file.close()

		@@ -289,76 +289,76 @@ plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red',
		set output 'csr-legacy-light2-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:4 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:4 title 'CSR Legacy Light2' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:5 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:5 title 'CSR Legacy Light2' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-light3-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:4 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:4 title 'CSR Legacy Light3' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:6 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:6 title 'CSR Legacy Light3' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-light4-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:4 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:4 title 'CSR Legacy Light4' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:7 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:7 title 'CSR Legacy Light4' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-light5-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:4 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:4 title 'CSR Legacy Light5' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:8 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:8 title 'CSR Legacy Light5' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-light6-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:4 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:4 title 'CSR Legacy Light6' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:9 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:9 title 'CSR Legacy Light6' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-light-without-atomic-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:5 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:5 title 'CSR Legacy LightWithoutAtomic' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:10 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:10 title 'CSR Legacy LightWithoutAtomic' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-scalar-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:6 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:6 title 'CSR Legacy Scalar' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:11 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:11 title 'CSR Legacy Scalar' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-vector-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:7 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:7 title 'CSR Legacy Vector' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:12 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:12 title 'CSR Legacy Vector' with lines linewidth 0.5 lt rgb 'green',
		set output 'csr-legacy-multivector-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:8 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:8 title 'CSR Legacy MultiVector' with lines linewidth 0.5 lt rgb 'green',
		'cusparse.gplt' using 1:13 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:13 title 'CSR Legacy MultiVector' with lines linewidth 0.5 lt rgb 'green',
		set output 'ellpack-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:9 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:9 title 'Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:10 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:10 title 'Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		'cusparse.gplt' using 1:14 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:14 title 'Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:15 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:15 title 'Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		set output 'sliced-ellpack-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:11 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:11 title 'Sliced Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:12 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:12 title 'Sliced Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		'cusparse.gplt' using 1:16 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:16 title 'Sliced Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:17 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:17 title 'Sliced Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		set output 'chunked-ellpack-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:13 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:13 title 'Chunked Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:14 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:14 title 'Chunked Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		'cusparse.gplt' using 1:18 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:18 title 'Chunked Ellpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:19 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:19 title 'Chunked Ellpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		set output 'bi-ellpack-vs-cusparse.eps'
		plot 'cusparse.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'cusparse.gplt' using 1:2 title 'cuSparse' with lines linewidth 0.5 lt rgb 'red', \
		'cusparse.gplt' using 1:15 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:15 title 'BiEllpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:16 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:16 title 'BiEllpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		'cusparse.gplt' using 1:20 title '' with dots linewidth 2 lt rgb 'green', \
		'cusparse.gplt' using 1:20 title 'BiEllpack' with lines linewidth 0.5 lt rgb 'green', \
		'cusparse.gplt' using 1:21 title '' with dots linewidth 2 lt rgb 'blue', \
		'cusparse.gplt' using 1:21 title 'BiEllpack Legacy' with lines linewidth 0.5 lt rgb 'blue'
		set output 'ellpack-vs-ellpack-legacy.eps'
		plot 'ellpack.gplt' using 1:2 title '' with dots linewidth 2 lt rgb 'red', \
		'ellpack.gplt' using 1:2 title 'Ellpack' with lines linewidth 0.5 lt rgb 'red', \