1143418754
[email protected]
Hi,
I am trying to solve a large linear equation, which needs a GPU solver as
comparison. I install a CUDA-enabled PETSc and petsc4py from sources using the
release tarball. According to the test results after installation, the PETSc
can successfully work with cuda.
All my programs are written in python, so I turn to petsc4py. But I do not find
any commands that define variables on coda device or define where the direct
solver is executed. I check `nvidia-smi` and find my cuda does not work at all
when executing my python script:
from petsc4py import PETSc
import numpy as np
n = 1000
nnz = 3 * np.ones(n, dtype=np.int32)
nnz[0] = nnz[-1] = 2
A = PETSc.Mat()
A.createAIJ([n, n], nnz=nnz)
# First set the first row
A.setValue(0, 0, 2)
A.setValue(0, 1, -1)
# Now we fill the last row
A.setValue(999, 998, -1)
A.setValue(999, 999, 2)
# And now everything else
for index in range(1, n - 1):
A.setValue(index, index - 1, -1)
A.setValue(index, index, 2)
A.setValue(index, index + 1, -1)
A.assemble()
indexptr, indices, data = A.getValuesCSR()
b = A.createVecLeft()
b.array[:] = 1
for i in range(10):
ksp = PETSc.KSP().create()
ksp.setOperators(A)
ksp.setType('preonly')
ksp.setConvergenceHistory()
ksp.getPC().setType('lu')
x = A.createVecRight()
ksp.solve(2*b, x)
residual = A * x - 2*b
if i % 10 == 0:
print(f"The relative residual is: {residual.norm() / b.norm()}.??)
What should I do to utlize GPU to execute the KSP task? Are there some settings
to be modified?
Looking forward to your early reply. Thanks a lot.
