[Bug] PTODSL's pto.simt_allreduce_sum/max/min rejects signless integer inputs #1580

NotImplementedError: all_reduce: unsupported dtype i32

from ptodsl import pto


@pto.jit(name="kernel", kernel_kind="vector", target="a5", mode="explicit")
def kernel(A: pto.ptr(pto.i32, "gm")):
    pto.init_core()
    with pto.simt(32, 1, 1):
        v = pto.cast(pto.get_tid_x(), pto.i32)
        r = pto.simt_allreduce_max(v, threads=32)
        pto.store(r, A, 0)


kernel.compile()
print("compiled")

Traceback (most recent call last):
  File "test_simt_allreduce_i32.py", line 14, in <module>
    kernel.compile()
  File "ptodsl/ptodsl/_jit.py", line 343, in compile
    return self._compiler.compile(**constexpr_bindings)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "ptodsl/ptodsl/_kernel_compilation.py", line 142, in compile
    compiled.build()
  File "ptodsl/ptodsl/_tracing/artifacts.py", line 36, in build
    built = self._module_factory()
            ^^^^^^^^^^^^^^^^^^^^^^
  File "ptodsl/ptodsl/_tracing/runtime.py", line 93, in build_module
    self.trace_entry(*args)
  File "ptodsl/ptodsl/_tracing/runtime.py", line 135, in trace_entry
    result = self._callback(*args, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "test_simt_allreduce_i32.py", line 10, in kernel
    r = pto.simt_allreduce_max(v, threads=32)
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "ptodsl/ptodsl/_allreduce.py", line 423, in simt_allreduce_sum
    return _simt_allreduce(value, threads=threads, scale=scale,
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "ptodsl/ptodsl/_allreduce.py", line 389, in _simt_allreduce
    raise NotImplementedError(f"all_reduce: unsupported dtype {raw_value.type}")
NotImplementedError: all_reduce: unsupported dtype i32