Skip to content

Commit 2ed752e

Browse files
committed
fix softmax tutorial
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 2cab9ed commit 2ed752e

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

python/tutorials/02-fused-softmax.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -175,14 +175,7 @@ def allocated_slm_size(size_smem):
175175
num_programs = n_rows
176176

177177
# Create a number of persistent programs.
178-
kernel[(num_programs, 1, 1)](
179-
y,
180-
x,
181-
x.stride(0),
182-
y.stride(0),
183-
n_rows,
184-
n_cols,
185-
)
178+
kernel[(num_programs, 1, 1)](y, x, x.stride(0), y.stride(0), n_rows, n_cols, BLOCK_SIZE)
186179
return y
187180

188181

0 commit comments

Comments
 (0)