twepp2015/data/throughput.py

import numpy as np
import matplotlib.pyplot as plt

gpu_data = np.loadtxt('throughput.gpu')
cpu_data = np.loadtxt('throughput.cpu')

plt.rc('font', **dict(family='serif'))

plt.figure(figsize=(8, 3))

plt.semilogx(gpu_data[:,0], gpu_data[:,1], '*-', color='#3b5b92', label='GPU')
plt.semilogx(cpu_data[:,0], cpu_data[:,1], 'o-', color='#d54d4d', label='CPU')
plt.xticks([1e4,1e6,1e8,1e10])
plt.yticks([0,2000,4000,6000,8000])

plt.xlabel('Data size in B')
plt.ylabel('Throughput in MB/s')
plt.legend(loc='lower right')
plt.savefig('throughput.pdf', dpi=300, bbox_inches='tight')