Feature request: Numpy-like array indexing

[kennykos]

In a pk.workunit, my code fails when I try to access a cupy array as x[a, b] instead of x[a][b]. For example, the script

import cupy as cp
import pykokkos as pk

@pk.workunit
def work(wid, a):
    a[wid,wid] = a[wid,wid] + 1

def main():
    N = 10
    a = cp.ones((N, N))
    pk.set_default_space(pk.Cuda)
    pk.parallel_for("work", 10, work, a=a)
    print(a)

main()

fails with error

Error on line 6 : 'a' is not a View1D
DEBUG AST:
Subscript(
    value=Name(id='a', ctx=Load()),
    slice=Tuple(
        elts=[
            Name(id='wid', ctx=Load()),
            Name(id='wid', ctx=Load())],
        ctx=Load()),
    ctx=Store())
    a[wid,wid] = a[wid,wid] + 1
    ^^^^^^^^^^
Translation of <ast.FunctionDef object at 0x150231e8fd00> work failed