Check for Python intitialization before GIL status

will-cromar · will-cromar · commit ce41ec852627 · 2023-01-26T17:16:38.000Z
diff --git a/torch_xla/csrc/xla_graph_executor.cpp b/torch_xla/csrc/xla_graph_executor.cpp
@@ -642,7 +642,7 @@ std::vector<at::Tensor> XLAGraphExecutor::GetTensorsFused(
   // completes. Release the GIL so other threads can proceed and unblock any
   // collective computations.
   PyThreadState* save = nullptr;
-  if (PyGILState_Check()) {
+  if (Py_IsInitialized() && PyGILState_Check()) {
     save = PyEval_SaveThread();
   }
   std::vector<xla::Literal> literals =

Original file line number	Diff line number	Diff line change
`@@ -642,7 +642,7 @@ std::vector<at::Tensor> XLAGraphExecutor::GetTensorsFused(`
`642`	`642`	`// completes. Release the GIL so other threads can proceed and unblock any`
`643`	`643`	`// collective computations.`
`644`	`644`	`PyThreadState* save = nullptr;`
`645`		`- if (PyGILState_Check()) {`
	`645`	`+ if (Py_IsInitialized() && PyGILState_Check()) {`
`646`	`646`	`save = PyEval_SaveThread();`
`647`	`647`	`}`
`648`	`648`	`std::vector<xla::Literal> literals =`