[REQ] Allow tiles to be passed to user functions #682
Description
Description
Permit tiles to be passed to wp.func functions. Eg
@wp.func
def tile_sum_func(a: wp.Tile(dtype=float, shape=(TILE_M, TILE_N), storage="shared")):
return wp.tile_sum(a) * 0.5