Skip to content

Commit 8be567e

Browse files
committed
Change view(-1) in ninetoothed_swiglu() to flatten() for handling non-contiguous inputs
1 parent 92900c9 commit 8be567e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

swiglu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ def swiglu_kernel(
2121

2222

2323
def ninetoothed_swiglu(a, b):
24-
a_1d = a.view(-1)
25-
b_1d = b.view(-1)
24+
a_1d = a.flatten()
25+
b_1d = b.flatten()
2626

2727
c = torch.empty_like(a_1d)
2828

0 commit comments

Comments
 (0)