code: def get_jit_fused_gelu_forward_func():
      from colossalai.kernel.jit.bias_gelu import bias_gelu
  
      def bloom_gelu_forward(
          x: torch.Tensor, bias: torch.Tensor
      ) -> torch.Tensor:
          return bias_gelu(bias, x)
  
      return bloom_gelu_forward
  
label_desc: Use image node_id 3 for calling a global function with example usage: get_jit_fused_gelu_forward_func() and returns: bloom_gelu_forward, bias_gelu