`core.fusions.fused_bias_gelu`#

Module Contents#

Classes#

Functions#

`bias_gelu`
`bias_gelu_back`

Data#

bias_gelu_impl

API#

core.fusions.fused_bias_gelu.bias_gelu(bias, y)#

core.fusions.fused_bias_gelu.bias_gelu_back(g, bias, y)#

class core.fusions.fused_bias_gelu.GeLUFunction#

Bases: torch.autograd.Function

static forward(ctx, input, bias)#

static backward(ctx, grad_output)#

classmethod apply(*args, **kwargs)#

core.fusions.fused_bias_gelu.bias_gelu_impl#: None