| 3537 | #endif // CUDA_VERSION >= 11000 |
| 3538 | |
| 3539 | port::StatusOr<std::unique_ptr<blas::IBlasLtMatmulPlan>> |
| 3540 | CUDABlas::CreateBlasLtMatmulPlan(const blas::BlasLtMatmulPlanParams& p) { |
| 3541 | #if CUDA_VERSION >= 11000 |
| 3542 | auto cuda_plan = std::make_unique<CUDABlasLtMatmulPlan>(); |
| 3543 | SE_RETURN_IF_ERROR(cuda_plan->init(p)); |
| 3544 | return static_cast<std::unique_ptr<blas::IBlasLtMatmulPlan>>( |
| 3545 | std::move(cuda_plan)); |
| 3546 | #else |
| 3547 | return port::Status( |
| 3548 | port::error::UNIMPLEMENTED, |
| 3549 | "CreateBlasLtMatmulPlan is not supported with this version of CUDA"); |
| 3550 | #endif |
| 3551 | } |
| 3552 | |
| 3553 | port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>> |
| 3554 | CUDABlas::GetBlasLtMatmulAlgorithmsInternal(const blas::IBlasLtMatmulPlan* plan, |
no test coverage detected