use opencl solver (for testing)
This commit is contained in:
parent
8daab636ea
commit
eb4d9be6cf
@ -10,6 +10,8 @@ import gridlock
|
||||
|
||||
from matplotlib import pyplot
|
||||
|
||||
from opencl_fdfd import cg_solver
|
||||
|
||||
__author__ = 'Jan Petykiewicz'
|
||||
|
||||
|
||||
@ -185,16 +187,25 @@ def test1():
|
||||
# pmcg.draw_cuboid(center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
|
||||
# pmcg.visualize_isosurface()
|
||||
|
||||
'''
|
||||
Solve!
|
||||
'''
|
||||
A = fdfd_tools.operators.e_full(omega, dxes,
|
||||
epsilon=vec(grid.grids),
|
||||
pec=vec(pecg.grids),
|
||||
pmc=vec(pmcg.grids)).tocsr()
|
||||
b = -1j * omega * vec(J)
|
||||
x = solve_A(A, b)
|
||||
# x = solve_A(A, b)
|
||||
|
||||
x = cg_solver(omega, dxes, J=vec(J), epsilon=vec(grid.grids),
|
||||
pec=vec(pecg.grids), pmc=vec(pmcg.grids))
|
||||
E = unvec(x, grid.shape)
|
||||
|
||||
print('Norm of the residual is ', numpy.linalg.norm(A @ x - b))
|
||||
|
||||
'''
|
||||
Plot results
|
||||
'''
|
||||
def pcolor(v):
|
||||
vmax = numpy.max(numpy.abs(v))
|
||||
pyplot.pcolor(v, cmap='seismic', vmin=-vmax, vmax=vmax)
|
||||
|
Loading…
Reference in New Issue
Block a user