Searched refs:NcclIsEnabled (Results 1 – 3 of 3) sorted by relevance
39 static bool NcclIsEnabled();
31 /* static */ bool NcclAllReduceThunk::NcclIsEnabled() { in NcclIsEnabled() function in xla::gpu::NcclAllReduceThunk
1392 << "; NCCL is enabled: " << NcclAllReduceThunk::NcclIsEnabled(); in HandleAllReduce()1421 crs->IsCrossReplicaAllReduce(), NcclAllReduceThunk::NcclIsEnabled()); in HandleAllReduce()