[Tags: v2.1] Error behavior for 'res_out = flag_gems.native_dropout(inp, p, True)'

[Dayrker]

Describe the bug

位置：
FlagGems/tests/test_special_ops.py::test_accuracy_dropout
FlagGems/src/flag_gems/ops/dropout.py
现象：

1. res_out = flag_gems.native_dropout(inp, p, True)使用的时候会报错
   mask0 = r0 > p
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.fp32')

[StrongSpoon]

Could you please provide a demo code to recurring this error?

[Dayrker]

Could you please provide a demo code to recurring this error?

change here.

then use "pytest tests/test_special_ops.py::test_accuracy_dropout"

[Dayrker]

I can also run normally with python terminal, like

But when using pytest, it will be that error.

[StrongSpoon]

This is due to the data type of operand p. Triton kernel can't accept p as numpy.float32 directly. Annotate the code "p = np.float32(p)" and this error disappears.

[Dayrker]

This is due to the data type of operand p. Triton kernel can't accept p as numpy.float32 directly. Annotate the code "p = np.float32(p)" and this error disappears.

Get it. Tanhks!