Update on "Add Python declaration of torch._C and torch._C._autograd …

…modules." Differential Revision: [D24761503](https://our.internmc.facebook.com/intern/diff/D24761503) [ghstack-poisoned]
pytorch · Nov 6, 2020 · e0bae07 · e0bae07
1 parent f1c2415
commit e0bae07
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/torch/csrc/distributed/autograd/init.cpp b/torch/csrc/distributed/autograd/init.cpp
@@ -23,8 +23,10 @@ PyObject* dist_autograd_init(PyObject* _unused, PyObject* noargs) {
   }
 
   auto torch_C_module = THPObjectPtr(PyImport_ImportModule("torch._C"));
-  if (!torch_C_module)
-    return nullptr;
+  if (!torch_C_module) {
+    throw python_error();
+  }
+
   auto torch_C_m = py::handle(torch_C_module).cast<py::module>();
   auto m = torch_C_m.def_submodule("_distributed_autograd", "distributed autograd bindings");