Method SetWorkspace

tensorflow/stream_executor/cuda/cuda_blas.cc:268–280

}

bool CUDABlas::SetWorkspace(Stream* stream, void *workspace, size_t workspaceSizeInBytes) {
  CHECK(blas_ != nullptr);
  if (!SetStream(stream)) {
    return false;
  }
  cublasStatus_t ret = cublasSetWorkspace(blas_, workspace, workspaceSizeInBytes);
  VLOG(2) << "set cublas workspace";
  if (ret != CUBLAS_STATUS_SUCCESS) {
    LOG(ERROR) << "failed to set cublas workspace: " << ToString(ret);
    return false;
  }
  return true;
}

bool CUDABlas::SetStream(Stream *stream) {
  CHECK(stream != nullptr);