IncompatibleTypeErrorImpl('invalid operands of type pointer<int64> and triton.language.int32')

[wuyouliaoxi]

Hello, I met this error when I used rotary positional encoding in flash attention.

The attribute rotary_emb_dim = int(embed_dim // num_heads // 2)

ncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')
apply_rotary(
File "/home/woody/iwi5/iwi5070h/software/private/conda/envs/mmseg/lib/python3.8/site-packages/flash_attn/ops/triton/rotary.py", line 202, in apply_rotary
raise CompilationError(fn.src, node, repr(e)) from e
triton.compiler.errors.CompilationError: at 32:36: # Meta-parameters
BLOCK_K: tl.constexpr,
IS_SEQLEN_OFFSETS_TENSOR: tl.constexpr,
IS_VARLEN: tl.constexpr,
INTERLEAVED: tl.constexpr,
CONJUGATE: tl.constexpr,
BLOCK_M: tl.constexpr,
):
pid_m = tl.program_id(axis=0)
pid_batch = tl.program_id(axis=1)
pid_head = tl.program_id(axis=2)
rotary_dim_half = rotary_dim // 2
^

My package versions are :
Name: triton
Version: 2.1.0

Name: flash-attn
Version: 2.7.0.post2