From 6ef4c63c4d31e2414647c134bd4a771dcd87b31f Mon Sep 17 00:00:00 2001
From: Francisco Massa <fvsmassa@gmail.com>
Date: Wed, 2 Jul 2025 10:12:47 +0000
Subject: [PATCH] Fix solution printing when problem is unsolvable

This is a followup from https://github.com/pytorch-labs/autoparallel/pull/21
---
 autoparallel/optimize_sharding.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/autoparallel/optimize_sharding.py b/autoparallel/optimize_sharding.py
index aa054ad0..decd321b 100644
--- a/autoparallel/optimize_sharding.py
+++ b/autoparallel/optimize_sharding.py
@@ -410,7 +410,7 @@ def get_solution(self, verbose=False):
         self.res = [k for k, v in sol.items() if v == 1]
 
         if self.prob.status == -1:
-            self.print()
+            print(self.get_log())
             raise RuntimeError("Unsolvable problem")
 
         opt = {}