Port indexing operation from CUDA backend

[dybber]

This will have to add additional parameters to all functions, in which the index arrays can be passed