gangliao · May 24, 2024 10:53 · gangliao · Mar 29, 2018 · abidmalikwaterloo · Mar 30, 2018
diff --git a/plot_bench_1.py b/plot_bench_1.py
 import matplotlib.pyplot as plt
 #for plotting
 import numpy as np 

 # create plot
 fig, ax = plt.subplots()
 bar_width = 0.15
 opacity = 0.8

 xlabel= np.array([8, 16, 32, 64])
 index = np.arange(4)
 tcpVals = [304, 255, 118, 83]
 ibVals = [304, 151, 92, 50]

 r1 = plt.bar(index, tcpVals, bar_width,
            alpha=opacity,
            color='#595959',
            label='TCP')
    
 r2 = plt.bar(index + bar_width, ibVals, bar_width,
            alpha=opacity,
            color='#F6921E',
            label='InfiniBand')

 ax.set_xticks(index + bar_width / 2)
 ax.set_xticklabels(xlabel)

 plt.xlabel('# GPUs')
 plt.ylabel('# Secs')
 ttlStr = 'Ring Allreduce on Acoustic Model (LSTM + WarpCTC)'
 plt.title(ttlStr)
 plt.legend()
    
 plt.tight_layout()
 plt.show()
diff --git a/run_ring_allreduce.md b/run_ring_allreduce.md
	import matplotlib.pyplot as plt
	#for plotting
	import numpy as np

	# create plot
	fig, ax = plt.subplots()
	bar_width = 0.15
	opacity = 0.8

	xlabel= np.array([8, 16, 32, 64])
	index = np.arange(4)
	tcpVals = [304, 255, 118, 83]
	ibVals = [304, 151, 92, 50]

	r1 = plt.bar(index, tcpVals, bar_width,
	alpha=opacity,
	color='#595959',
	label='TCP')

	r2 = plt.bar(index + bar_width, ibVals, bar_width,
	alpha=opacity,
	color='#F6921E',
	label='InfiniBand')

	ax.set_xticks(index + bar_width / 2)
	ax.set_xticklabels(xlabel)

	plt.xlabel('# GPUs')
	plt.ylabel('# Secs')
	ttlStr = 'Ring Allreduce on Acoustic Model (LSTM + WarpCTC)'
	plt.title(ttlStr)
	plt.legend()

	plt.tight_layout()
	plt.show()