Can‘t solve QP problem when dimension is too high.

[WHQ1111]

trafficstars

When I try this:

G = torch.eye(2000).expand(8,2000,2000).cuda() e = torch.randint(0,1,[8,2000]).cuda() C = torch.eye(2000).expand(8,2000,2000).cuda() h = torch.randint(0,1,[8,2000]).cuda() A = torch.randn(8,100,2000).cuda() b = torch.zeros(8,100).cuda() qp_sol = QPFunction(verbose=False, maxIter=15)(G, e, C, h, A, b)

I get

RuntimeError: CUDA error: invalid configuration argument

How to solve this problem? Or it can't work when dimension is too high.

[Tsingularity]

i have the same error here.

Have u solved this yet?

[nothingeasy]

I found the problem may cause by pre_factor_kkt function where use lu_solve(*Q_LU) function in solvers/pdipd/batch.py, which indicate torch.lu_solve can not solve this part.

[LinkToPast1900]

https://github.com/pytorch/pytorch/pull/61815