Skip to content

Commit 906a166

Browse files
committed
Rename ninetoothed_swiglu to swiglu
1 parent da33f58 commit 906a166

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

swiglu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def swiglu_kernel(
2020
c = a * gate # noqa: F841
2121

2222

23-
def ninetoothed_swiglu(a, b):
23+
def swiglu(a, b):
2424
a_1d = a.flatten()
2525
b_1d = b.flatten()
2626

@@ -80,7 +80,7 @@ def torch_swiglu(
8080
b = torch.rand(shape, dtype=dtype, device=device)
8181
c = torch.rand(shape, dtype=dtype, device=device)
8282

83-
ninetoothed_output = ninetoothed_swiglu(a, b)
83+
ninetoothed_output = swiglu(a, b)
8484
torch_output = torch_swiglu(a, b)
8585
triton_output = triton_swiglu(a, b)
8686
print(ninetoothed_output)
@@ -121,7 +121,7 @@ def benchmark(m, n, provider):
121121

122122
if provider == "ninetoothed":
123123
ms, min_ms, max_ms = triton.testing.do_bench(
124-
lambda: ninetoothed_swiglu(a, b), quantiles=quantiles
124+
lambda: swiglu(a, b), quantiles=quantiles
125125
)
126126
elif provider == "torch":
127127
ms, min_ms, max_ms = triton.testing.do_bench(

0 commit comments

Comments
 (0)