code: def load_sru_mod():
      global SRU_FWD_FUNC, SRU_BWD_FUNC, SRU_BiFWD_FUNC, SRU_BiBWD_FUNC
      global SRU_STREAM
      if check_sru_requirement():
          from cupy.cuda import function
          from pynvrtc.compiler import Program
  
          # This sets up device to use.
          device = torch.device("cuda")
          tmp_ = torch.rand(1, 1).to(device)
  
          sru_prog = Program(
              SRU_CODE.encode("utf-8"), "sru_prog.cu".encode("utf-8")
          )
          sru_ptx = sru_prog.compile()
          sru_mod = function.Module()
          sru_mod.load(bytes(sru_ptx.encode()))
  
          SRU_FWD_FUNC = sru_mod.get_function("sru_fwd")
          SRU_BWD_FUNC = sru_mod.get_function("sru_bwd")
          SRU_BiFWD_FUNC = sru_mod.get_function("sru_bi_fwd")
          SRU_BiBWD_FUNC = sru_mod.get_function("sru_bi_bwd")
  
          stream = namedtuple("Stream", ["ptr"])
          SRU_STREAM = stream(
              ptr=torch.cuda.current_stream().cuda_stream
          )
  
imports: import subprocess
  import platform
  import os
  import re
  import configargparse
  import torch
  import torch.nn
  from torch.autograd import Function
  from torch.cuda.amp import custom_fwd, custom_bwd
  from collections import namedtuple
label_desc: Use image node_id 2 for calling a global function with example usage: load_sru_mod() without return types