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

[WHQ1111]

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]

pytorch/pytorch#61815