Searched refs:max_algorithm_count (Results 1 – 5 of 5) sorted by relevance
110 std::optional<int> max_algorithm_count) { in GetPlanAndAlgorithms() argument116 if (!max_algorithm_count) max_algorithm_count = max_autotune_algorithm_count; in GetPlanAndAlgorithms()188 blas_lt->GetMatmulAlgorithms(plan, preference, *max_algorithm_count)); in GetPlanAndAlgorithms()
428 std::optional<int> max_algorithm_count;429 if (!use_autotune) max_algorithm_count = 1;432 GetPlanAndAlgorithms(stream, matmul_params, max_algorithm_count);
84 std::optional<int> max_algorithm_count = std::nullopt);
229 size_t max_algorithm_count) { in GetMatmulAlgorithms() argument230 max_algorithm_count = std::min(max_algorithm_count, size_t{INT_MAX}); in GetMatmulAlgorithms()231 std::vector<cublasLtMatmulHeuristicResult_t> results(max_algorithm_count); in GetMatmulAlgorithms()242 preference.get(), max_algorithm_count, results.data(), in GetMatmulAlgorithms()
151 size_t max_algorithm_count = 128);