Register resource GPU kernels for NextIteration and Exit.
Prior to this change, accessing a GPU resource variable from a while loop would fail, because the resource NextIteration and Exit nodes would cause the placer to put the whole loop body on CPU, including variable reads, etc. which must be colocated with the resource. PiperOrigin-RevId: 226210621
Loading
Please sign in to comment