Estou tentando criar um invólucro pycuda inspirado na biblioteca scikits-cuda para algumas operações fornecidas na nova biblioteca cuSolver da Nvidia. Quero resolver um sistema linear do formato AX = B por fatoração LU, para executar esse primeiro uso docublasSgetrfBatched método de scikits-cuda, que me fornece a fatoração LU; então com essa fatoração eu quero resolver o sistema usandocusolverDnSgetrs do cuSolve que eu quero agrupar, quando executo o status de retorno da computação 3, as matrizes que pretendem me dar a resposta não mudam, MAS o * devInfo é zero, procurando na documentação do cusolver:

CUSOLVER_STATUS_INVALID_VALUE = Um valor ou parâmetro não suportado foi passado para a função (um tamanho de vetor negativo, por exemplo).

libcusolver.cusolverDnSgetrs.restype=int
libcusolver.cusolverDnSgetrs.argtypes=[_types.handle,
                                   ctypes.c_char,
                                   ctypes.c_int,
                                   ctypes.c_int,
                                   ctypes.c_void_p,
                                   ctypes.c_int,
                                   ctypes.c_void_p,
                                   ctypes.c_void_p,
                                   ctypes.c_int,
                                   ctypes.c_void_p]

"""
handle is the handle pointer given by calling cusolverDnCreate() from cuSolver
LU is the LU factoriced matrix given by cublasSgetrfBatched() from scikits
P is the pivots matrix given by cublasSgetrfBatched()
B is the right hand matix from AX=B
"""
def cusolverSolveLU(handle,LU,P,B):
   rows_LU ,cols_LU = LU.shape
   rows_B, cols_B = B.shape
   B_gpu = gpuarray.to_gpu(B.astype('float32'))
   info_gpu = gpuarray.zeros(1, np.int32)

   status=libcusolver.cusolverDnSgetrs(
               handle, 'n', rows_LU, cols_B,
               int(LU.gpudata), cols_LU,
               int(P.gpudata), int(B_gpu.gpudata),
               cols_B, int(info_gpu.gpudata))
   print info_gpu   
   print status

handle= cusolverCreate() #get the initialization of cusolver
LU, P = cublasLUFactorization(...)
B = np.asarray(np.random.rand(3, 3), np.float32)
cusolverSolveLU(handle,LU,P,B)

A saída:

[0]

3

O que estou fazendo de errado?