code: def _get_cpp_module():
      """
      JIT compiles the cpp_spanning_tree module.
      """
      global _cpp_module
      if _cpp_module is None:
          import os
  
          from torch.utils.cpp_extension import load
  
          path = os.path.join(
              os.path.dirname(os.path.abspath(__file__)),
              "spanning_tree.cpp",
          )
          _cpp_module = load(
              name="cpp_spanning_tree",
              sources=[path],
              extra_cflags=["-O2"],
              verbose=True,
          )
      return _cpp_module
  
imports: import itertools
  import math
  import torch
  from torch.distributions import constraints
  from torch.distributions.utils import lazy_property
  from pyro.distributions.torch_distribution import TorchDistribution
label_desc: Use image node_id 1 for calling a global function with example usage: _get_cpp_module() and returns: _cpp_module