| 3607 | } |
| 3608 | |
| 3609 | port::StatusOr<std::vector<std::unique_ptr<blas::IBlasLtMatmulAlgorithm>>> |
| 3610 | CUDABlas::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan* plan, |
| 3611 | size_t max_workspace_size, |
| 3612 | int max_algorithm_count) { |
| 3613 | return GetBlasLtMatmulAlgorithmsInternal(plan, max_workspace_size, |
| 3614 | max_algorithm_count); |
| 3615 | } |
| 3616 | |
| 3617 | #if CUDA_VERSION >= 11000 |
| 3618 | bool CUDABlas::DoBlasLtMatmulInternal( |
no outgoing calls
no test coverage detected