Adds some quick timing code for CPU and GPU code