Makes the appropriate calls to FormulateAndSolveMIPFromProblem().

tensorflower-gardener · tensorflower-gardener · commit 5194fa6071f0 · 2025-06-02T11:57:37.000-07:00
PiperOrigin-RevId: 766270584
diff --git a/third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc b/third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc
@@ -1997,9 +1997,7 @@ CreateAutoShardingSolverRequestAndCallSolver(
   }
 
   const auto converted_problem = ConvertToProblem(request);
-  const auto converted_request = ConvertToSolverRequest(converted_problem);
-  return FormulateAndSolveMIPFromSolverRequest(converted_request,
-                                               GetParams(request));
+  return FormulateAndSolveMIPFromProblem(converted_problem, GetParams(request));
 }
 
 void CheckHloSharding(

Original file line number	Diff line number	Diff line change
`@@ -1997,9 +1997,7 @@ CreateAutoShardingSolverRequestAndCallSolver(`
`1997`	`1997`	`}`
`1998`	`1998`
`1999`	`1999`	`const auto converted_problem = ConvertToProblem(request);`
`2000`		`- const auto converted_request = ConvertToSolverRequest(converted_problem);`
`2001`		`- return FormulateAndSolveMIPFromSolverRequest(converted_request,`
`2002`		`- GetParams(request));`
	`2000`	`+ return FormulateAndSolveMIPFromProblem(converted_problem, GetParams(request));`
`2003`	`2001`	`}`
`2004`	`2002`
`2005`	`2003`	`void CheckHloSharding(`