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

Method SetStream

tensorflow/stream_executor/rocm/rocm_rng.cc:142–151  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

140}
141
142bool GpuRng::SetStream(Stream* stream) {
143 hiprandStatus_t ret =
144 wrap::hiprandSetStream(parent_, rng_, AsGpuStreamValue(stream));
145 if (ret != HIPRAND_STATUS_SUCCESS) {
146 LOG(ERROR) << "failed to set stream for random generation: " << ret;
147 return false;
148 }
149
150 return true;
151}
152
153// Returns true if std::complex stores its contents as two consecutive
154// elements. Tests int, float and double, as the last two are independent

Callers

nothing calls this directly

Calls 1

AsGpuStreamValueFunction · 0.85

Tested by

no test coverage detected