Cleaning up gpu_utils