How to use the accelerate.cuda.blas.Blas function in accelerate
To help you get started, we’ve selected a few accelerate examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
scivision / python-performance / matmul / matmul_pycuda.py View on Github 
def gemm(A, B, dD):
N = A.shape[0] # square matrices
'''
Note that all arrays are in Fortran order.
'''
# cuBLAS
blas = Blas()
start = timer()
blas.gemm('N', 'N', N, N, N, 1.0, A, B, 1.0, dD)
cuda_time = timer() - start
D = dD.copy_to_host()
print("CUBLAS took %f seconds" % cuda_time)
diff = np.abs(D - E)
print("Maximum error %f" % np.max(diff))
return D
