code: def load_spatial_module():
      global spatial_cuda_module
      if spatial_cuda_module is None:
          spatial_cuda_module = (
              op_builder.SpatialInferenceBuilder().load()
          )
      return spatial_cuda_module
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/ops/transformer/inference/diffusers_transformer_block.py
imports: import torch
  import torch.nn
  from ...None import op_builder
  from deepspeed import module_inject
  from .diffusers_attention import DeepSpeedDiffusersAttention
  from .bias_add import nhwc_bias_add
  from .diffusers_2d_transformer import Diffusers2DTransformerConfig
label_desc: Use image node_id 2 for calling a global function with example usage: load_spatial_module() and returns: spatial_cuda_module