[PyCUDA] gpuarray, return index of max