reduced cudaMallocs. I think this was slowing things down