I mean, just remove the cudaThreadSyncronize() as is currently done in this PR.