MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / OpenNcclChannels

Function OpenNcclChannels

tensorflow/compiler/xla/tests/collective_ops_test.cc:162–164  ·  view source on GitHub ↗

Shorter alias for this function.

Source from the content-addressed store, hash-verified

160
161// Shorter alias for this function.
162absl::flat_hash_set<GlobalDeviceId> OpenNcclChannels() {
163 return gpu::NcclAllReduceThunk::DevicesWithOpenNcclChannels();
164}
165
166template <typename T>
167static Eigen::half ToHalf(T value) {

Callers 1

XLA_TEST_FFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected