Remove the unused n_rows parameter in the triton_softmax_kernel function

Author: voltjia

code_size_comparison.py

@@ -7,7 +7,7 @@
 plt.rcParams["axes.labelweight"] = "bold"
 
 kernels = ("add", "softmax", "matmul", "conv2d", "attention")
-lines_of_code = {"Triton": (19, 26, 57, 110, 98), "NineToothed": (10, 12, 34, 17, 51)}
+lines_of_code = {"Triton": (19, 25, 57, 110, 98), "NineToothed": (10, 12, 34, 17, 51)}
 
 x = np.arange(len(kernels))
 width = 0.4

softmax.py

@@ -33,7 +33,6 @@ def triton_softmax_kernel(
     output_ptr,
     input_row_stride,
     output_row_stride,
-    n_rows,
     n_cols,
     BLOCK_SIZE: tl.constexpr,
 ):
@@ -63,7 +62,6 @@ def triton_softmax(input):
         output,
         input.stride(0),
         output.stride(0),
-        input.shape[0],
         input.shape[1],
         BLOCK_SIZE=triton.next_power_of_2(input.shape[-1]),
     )