blis_gemm_matrix_multiplication_with_numpy_verification.py

1import numpy
2from blis.py import gemm
3
4# Create two random float32 matrices
5A = numpy.random.uniform(size=(10, 20)).astype("float32")
6B = numpy.random.uniform(size=(20, 30)).astype("float32")
7
8# Compute matrix multiplication using blis
9# Signature: gemm(A, B, out=None, trans1=False, trans2=False)
10C = gemm(A, B)
11
12# Verify results with numpy
13expected = numpy.dot(A, B)
14numpy.testing.assert_allclose(C, expected, atol=1e-5)
15
16print("Matrix multiplication successful!")
17print(f"Result shape: {C.shape}")