Fix device flags

Change CUDA device flag from cudaDeviceScheduleYield to cudaDeviceScheduleSpin