[Tags: v2.1] Error behavior for 'flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)'

[Dayrker]

Describe the bug

位置：
FlagGems/tests/test_special_ops.py::test_pad
FlagGems/src/flag_gems/ops/pad.py
现象:

1. res_out = flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)使用会报错
   E cond = (dst_index_0 >= valid_dim0_start and dst_index_0 < valid_dim0_end)
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')

[StrongSpoon]

Please provide a minimal reproduction to help us solving this problem.

[Bowen12992]

@MARD1NO hey, please hava a look

[MARD1NO]

Describe the bug

位置： FlagGems/tests/test_special_ops.py::test_pad FlagGems/src/flag_gems/ops/pad.py 现象:

1. res_out = flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)使用会报错
   E cond = (dst_index_0 >= valid_dim0_start and dst_index_0 < valid_dim0_end)
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')

方便提供下 python triton版本么

我这里py311 triton2.3.1运行正常

[Dayrker]

Describe the bug

位置： FlagGems/tests/test_special_ops.py::test_pad FlagGems/src/flag_gems/ops/pad.py 现象:

1. res_out = flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)使用会报错
   E cond = (dst_index_0 >= valid_dim0_start and dst_index_0 < valid_dim0_end)
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')

方便提供下 python triton版本么

我这里py311 triton2.3.1运行正常

在这里替换使用flag_gems.pad并pytest才会报错。
python: 3.10.14
triton: 2.3.1

[MARD1NO]

Describe the bug

位置： FlagGems/tests/test_special_ops.py::test_pad FlagGems/src/flag_gems/ops/pad.py 现象:

1. res_out = flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)使用会报错
   E cond = (dst_index_0 >= valid_dim0_start and dst_index_0 < valid_dim0_end)
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')

方便提供下 python triton版本么
我这里py311 triton2.3.1运行正常

在这里替换使用flag_gems.pad并pytest才会报错。 python: 3.10.14 triton: 2.3.1

这里不能等价调用，最终我们实现替换的aten的pad，而一些参数预处理是会在 functional python层面处理，最后再送到 pad算子里

[Dayrker]

Describe the bug

位置： FlagGems/tests/test_special_ops.py::test_pad FlagGems/src/flag_gems/ops/pad.py 现象:

1. res_out = flag_gems.pad(ref_x, pad_params, pad_mode, pad_value)使用会报错
   E cond = (dst_index_0 >= valid_dim0_start and dst_index_0 < valid_dim0_end)
   E ^
   E IncompatibleTypeErrorImpl('invalid operands of type pointer and triton.language.int32')

方便提供下 python triton版本么
我这里py311 triton2.3.1运行正常

在这里替换使用flag_gems.pad并pytest才会报错。 python: 3.10.14 triton: 2.3.1

这里不能等价调用，最终我们实现替换的aten的pad，而一些参数预处理是会在 functional python层面处理，最后再送到 pad算子里

了解，多谢~
再请教一下，对这个case如果想使用这种形式的话，可以对输入or参数做什么样的处理使他能够跑起来呢。因为框架原因，这边没办法使用aten替换的形式来使用flaggems