src/transformers/models/bloom/modeling_bloom.py/GeLUFunction/forward
class GeLUFunction:    def forward(ctx, input: torch.Tensor) -> torch.Tensor:
        ctx.save_for_backward(input)
        return bloom_gelu_forward(input)