code: def get_jit_fused_dropout_add_func():
      from colossalai.kernel.jit import (
          bias_dropout_add_fused_inference,
          bias_dropout_add_fused_train,
      )
  
      def self_dropout_add(
          self,
          x: torch.Tensor,
          residual: torch.Tensor,
          prob: float,
          training: bool,
      ) -> torch.Tensor:
          bias = torch.zeros_like(x)
          if training:
              return bias_dropout_add_fused_train(
                  x, bias, residual, prob
              )
          return bias_dropout_add_fused_inference(
              x, bias, residual, prob
          )
  
      return self_dropout_add
  
label_desc: Use image node_id 2 for calling a global function with example usage: get_jit_fused_dropout_add_func() and returns: self_dropout_add, bias_dropout_add_fused_inference, bias_dropout_add_fused_train