| 332 | } |
| 333 | |
| 334 | bool ROCMBlas::SetStream(Stream *stream) { |
| 335 | CHECK(stream != nullptr); |
| 336 | CHECK(AsGpuStreamValue(stream) != nullptr); |
| 337 | CHECK(blas_ != nullptr); |
| 338 | rocblas_status ret = |
| 339 | wrap::rocblas_set_stream(parent_, blas_, AsGpuStreamValue(stream)); |
| 340 | if (ret != rocblas_status_success) { |
| 341 | LOG(ERROR) << "failed to set stream for rocBLAS calls: " << ToString(ret); |
| 342 | return false; |
| 343 | } |
| 344 | |
| 345 | return true; |
| 346 | } |
| 347 | |
| 348 | namespace { |
| 349 |
nothing calls this directly
no test coverage detected