We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 92900c9 commit 8be567eCopy full SHA for 8be567e
1 file changed
swiglu.py
@@ -21,8 +21,8 @@ def swiglu_kernel(
21
22
23
def ninetoothed_swiglu(a, b):
24
- a_1d = a.view(-1)
25
- b_1d = b.view(-1)
+ a_1d = a.flatten()
+ b_1d = b.flatten()
26
27
c = torch.empty_like(a_1d)
28
0 commit comments