warp icon indicating copy to clipboard operation
warp copied to clipboard

[REQ] Support for Python slicing syntax

Open mmacklin opened this issue 11 months ago • 0 comments

Description

Users should be able to select sub-ranges of arrays and tiles inside kernels, e.g.:

t = wp.tile_load(arr[i:i+m, j:j+n])
t = wp.tile_load(arr[i:i+m])

Context

More flexible indexing into arrays would allow writing e.g.: segmented scan with Warp tile kernels.

mmacklin avatar Feb 06 '25 01:02 mmacklin