blis_gemm_matrix_multiplication_with_numpy_arrays.py

1import numpy
2from blis.py import gemm
3
4# Create two sample matrices
5A = numpy.ones((16, 8), dtype='float32')
6B = numpy.ones((8, 16), dtype='float32')
7C = numpy.zeros((16, 16), dtype='float32')
8
9# Perform matrix multiplication: C = A @ B
10# gemm(A, B, out=C, trans1=False, trans2=False)
11gemm(A, B, out=C)
12
13print(C)