Method Init

tensorflow/stream_executor/cuda/cuda_blas.cc:225–243

#endif  // CUDA_VERSION >= 9000

bool CUDABlas::Init() {
  gpu::ScopedActivateExecutorContext sac{parent_};
  cublasStatus_t ret = cublasCreate(&blas_);
  if (ret != CUBLAS_STATUS_SUCCESS) {
    LOG(ERROR) << "failed to create cublas handle: " << ToString(ret);
    return false;
  }

#if CUDA_VERSION >= 11000
  ret = cublasLtCreate(&blasLt_);
  if (ret != CUBLAS_STATUS_SUCCESS) {
    LOG(ERROR) << "failed to create cublasLt handle: " << ToString(ret);
    return false;
  }
  LOG(INFO) << "Create cuBLASLt success.";
#endif  // CUDA_VERSION >= 11000

  return true;
}

CUDABlas::CUDABlas(gpu::GpuExecutor* parent)
    : parent_(CHECK_NOTNULL(parent)),