Remove unused const (#2714)
* remove unused const * accidentally commited benchmark plot
This commit is contained in:
@@ -1,5 +1,4 @@
|
|||||||
"""
|
"""Module for definition of GEGLU Triton kernels.
|
||||||
Module for definition of GEGLU Triton kernels.
|
|
||||||
|
|
||||||
See "GLU Variants Improve Transformer" (https://arxiv.org/abs/2002.05202).
|
See "GLU Variants Improve Transformer" (https://arxiv.org/abs/2002.05202).
|
||||||
|
|
||||||
@@ -12,8 +11,6 @@ import torch
|
|||||||
import triton
|
import triton
|
||||||
import triton.language as tl
|
import triton.language as tl
|
||||||
|
|
||||||
SQRT_2_PI: tl.constexpr = 0.7978845608028654 # sqrt(2/π)
|
|
||||||
|
|
||||||
|
|
||||||
@triton.jit
|
@triton.jit
|
||||||
def _geglu_fwd_kernel(
|
def _geglu_fwd_kernel(
|
||||||
|
|||||||
Reference in New Issue
Block a user