Commit f25abbfb authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Minor cleanup: remove unnecessary GetCudaContext.

Note that there is no protection against a caller of CUDAExecutor::Launch from
accidentally passing a Stream associated with the wrong CUDAExecutor.

This is no different from any other CUDAExecutor methods that take a Stream
argument, where we similarly have no such protection.

The main caller is Stream::ThenLaunch, which necessarily calls Launch on the
correct corresponding CUDAExecutor. Other callers use a similar pattern.

PiperOrigin-RevId: 176719918
parent 58d0e5b6
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment