Bug Description

Repro:

import warp as wp

wp.init()

@wp.kernel
def compute():

    t = wp.tile_ones(64, dtype=float)
    sum = wp.tile_sum(t)
    count = sum[0]

    print(count)

wp.launch_tiled(compute, dim=1, block_dim=64)

Error message:

/home/zcorse/Repos/warp/warp/native/tile_reduce.h(436): error: no operator "[]" matches these operands
operand types are: wp::tile_shared_t<wp::float32, wp::tile_layout_strided_t<wp::tile_tuple_t<1>, wp::tile_tuple_t<1>>, true>::Storage [ int ]
scratch = adj_ret.data[0];
^
/home/zcorse/Repos/warp/warp/native/tile_reduce.h(436): note #3328-D: built-in operator[](, <ptrdiff_t>) does not match because argument #1 does not match parameter
scratch = adj_ret.data[0];

System Information

No response