From 22004eb92ca2d4e6132749e351fa22b87dcaae5a Mon Sep 17 00:00:00 2001 From: xla authors Date: Fri, 27 Sep 2024 11:13:52 -0700 Subject: [PATCH] Removes the scaling coefficient for our solver-specific parameter `max_deterministic_time`. PiperOrigin-RevId: 679659247 --- xla/hlo/experimental/auto_sharding/auto_sharding_solver.cc | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/xla/hlo/experimental/auto_sharding/auto_sharding_solver.cc b/xla/hlo/experimental/auto_sharding/auto_sharding_solver.cc index c8321348ab7c2..8fca1bc7b81ab 100644 --- a/xla/hlo/experimental/auto_sharding/auto_sharding_solver.cc +++ b/xla/hlo/experimental/auto_sharding/auto_sharding_solver.cc @@ -427,9 +427,8 @@ AutoShardingSolverResult CallORToolsSolver( ",share_binary_clauses:false,random_seed:1,interleave_search:true"); } if (request.has_solver_timeout()) { - absl::StrAppend( - &solver_parameter_str, ",max_deterministic_time:", - 0.1 * request.solver_timeout().solver_timeout_in_seconds()); + absl::StrAppend(&solver_parameter_str, ",max_deterministic_time:", + request.solver_timeout().solver_timeout_in_seconds()); } solver->SetSolverSpecificParametersAsString(solver_parameter_str); }