| 143 | public: |
| 144 | BlasPlansAutotuneCache() {} |
| 145 | bool Find(const se::BatchMatmulParameters& params, |
| 146 | se::blas::AlgorithmConfig* config) const { |
| 147 | absl::MutexLock lock(&mu_); |
| 148 | auto iter = blas_plans_algorithms_map_.find(params); |
| 149 | if (iter == blas_plans_algorithms_map_.end()) { |
| 150 | return false; |
| 151 | } |
| 152 | *config = iter->second; |
| 153 | return true; |
| 154 | } |
| 155 | void Insert(const se::BatchMatmulParameters& params, |
| 156 | const se::blas::AlgorithmConfig& config) { |
| 157 | absl::MutexLock lock(&mu_); |
no test coverage detected