Skip to content

Commit 3e2c03b

Browse files
add the benchmark example
1 parent a37eeca commit 3e2c03b

File tree

2 files changed

+8
-15
lines changed

2 files changed

+8
-15
lines changed

PyLinearSolver/IterativeSolvers.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,11 @@
11
try:
22
from julia.api import Julia
33
from julia import Main
4+
5+
julia = Julia()
6+
Main.eval("using IterativeSolvers")
47
except:
58
print("Installing Julia is Required")
6-
7-
def start():
8-
julia = Julia()
9-
Main.using("IterativeSolvers")
109

1110
def cg(A,b,**kwargs):
1211
'''

examples/benchmark_scipy.py

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -10,24 +10,18 @@
1010
A = np.random.rand(n,n)
1111
A = A * A.T
1212
b = np.random.rand(n)
13-
IterativeSolvers.start()
1413

1514
start = time.time()
1615
x = IterativeSolvers.cg(A,b)
17-
print("Iterative Solvers CG Computational Time: %.3f"%(time.time()-start))
16+
print("Iterative Solvers CG Computational Time: %.7f"%(time.time()-start))
1817
print("Iterative Solvers CG Solution: ",x)
1918

2019
start = time.time()
2120
x = linalg.cg(A,b)
22-
print("Scipy CG Computational Time: %.3f"%(time.time()-start))
21+
print("Scipy CG Computational Time: %.7f"%(time.time()-start))
2322
print("Scipy CG Solution: ",x)
2423

2524
start = time.time()
26-
x = IterativeSolvers.cg(A,b)
27-
print("Iterative Solvers CG Computational Time: %.3f"%(time.time()-start))
28-
print("Iterative Solvers CG Solution: ",x)
29-
# N= 10
30-
31-
# N = 100
32-
33-
# N = 1000
25+
x = IterativeSolvers.gmres(A+2,b)
26+
print("Iterative Solvers CG Computational Time: %.7f"%(time.time()-start))
27+
print("Iterative Solvers CG Solution: ",x)

0 commit comments

Comments
 (0)