when I use flash_attn, it reports a segment error

[huangjch526]

Can you help me? I installed flash-attn 2.6.3 and cd csrc/layer_norm && pip install . in order to use InternVideo2, but when I in: from flash_attn.ops.rms_norm import DropoutAddRMSNorm I meet import dropout_layer_norm it reports a segment error and I've spent a day trying to find out what the cause is.

[yinanhe]

Sorry, we are not clear about this problem. You may need to ask the author of flash-attn. Or you can try reinstalling first.