ray_tpu: Catch preemption better, don't hang if there's a c level abort

ray_tpu: Catch preemption better, don't hang if there's a c level abort #4189