fix fatals when 'max_seqlen_q' is 'numpy.int32'

[sgjzfzzf]

PR Category

Operator

Type of Change

Bug Fix

Description

max_seqlen_q could be np.int32, but it only checks whether it's torch.Tensor now.

Issue

PR #781, #789

Progress

• Change is properly reviewed (1 reviewer required, 2 recommended).
• Change is responded to an issue.
• Change is fully covered by a UT.

Performance

[iclementine]

triton jit function 目前对于参数的处理方式非常 ad-hoc. numpy scalar 和 python scalar 是不同的东西，需要额外的逻辑才能处理。