floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

benchmark/scripts/SCIPY/ssyrk.py: Fix PEP8 issues.

This commit is contained in:
John Kirkham 2016-01-19 15:06:17 -05:00
parent cbb6649e97
commit adfa0ab878
1 changed files with 36 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
benchmark/scripts/SCIPY/ssyrk.py
View File

@ -8,32 +8,32 @@ from numpy import zeros
from numpy.random import randn from numpy.random import randn
from scipy.linalg import blas from scipy.linalg import blas
def run_ssyrk(N,l):
A = randn(N,N).astype('float32') def run_ssyrk(N, l):
C = zeros((N,N), dtype='float32')
A = randn(N, N).astype('float32')
C = zeros((N, N), dtype='float32')
start = time.time() start = time.time()
for i in range(0,l): for i in range(0, l):
C[...] = blas.ssyrk(1.0,A) C[...] = blas.ssyrk(1.0, A)
end = time.time() end = time.time()
timediff = (end -start) timediff = (end - start)
mflops = ( N*N*N) *l / timediff mflops = (N * N * N) * l / timediff
mflops *= 1e-6 mflops *= 1e-6
size = "%dx%d" % (N,N) size = "%dx%d" % (N, N)
print("%14s :\t%20f MFlops\t%20f sec" % (size,mflops,timediff)) print("%14s :\t%20f MFlops\t%20f sec" % (size, mflops, timediff))
if __name__ == "__main__": if __name__ == "__main__":
N=128 N = 128
NMAX=2048 NMAX = 2048
NINC=128 NINC = 128
LOOPS=1 LOOPS = 1
z=0 z = 0
for arg in sys.argv: for arg in sys.argv:
if z == 1: if z == 1:
N = int(arg) N = int(arg)
@ -54,6 +54,5 @@ if __name__ == "__main__":
print("From: %d To: %d Step=%d Loops=%d" % (N, NMAX, NINC, LOOPS)) print("From: %d To: %d Step=%d Loops=%d" % (N, NMAX, NINC, LOOPS))
print("\tSIZE\t\t\tFlops\t\t\t\t\tTime") print("\tSIZE\t\t\tFlops\t\t\t\t\tTime")
for i in range (N,NMAX+NINC,NINC): for i in range(N, NMAX + NINC, NINC):
run_ssyrk(i,LOOPS) run_ssyrk(i, LOOPS)