@guvectorize(['void(int32[:], int32[:])'], '(n)->()', target='cuda')def sum_row(inp, out): tmp = 0. for i in range(inp.shape[0]): tmp += inp[i] out[0] = tmp