@triton.jit
def gelu_backward(x):
    x2 = x * x
    tanh_ = tanh(_kAlpha * x * (1 + 0.044715 * x2))
    dx = 0.5 * (x * (1 - tanh_ * tanh_) * (0.1070322244089 * x2 + 
        0.797884560802865) + tanh_ + 1)
    return dx
