Commit 40874676 authored by Yuefeng Zhou's avatar Yuefeng Zhou Committed by TensorFlower Gardener
Browse files

Make nccl work in eager mode: wrap nccl ops in a defun; remove control... 

Make nccl work in eager mode: wrap nccl ops in a defun; remove control dependencies on NcclAllReduce

PiperOrigin-RevId: 228801431
parent d2490b91
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment