27 lines
540 B
Python
27 lines
540 B
Python
# m1 = MatrixMPI(numpy.random.uniform(0, 1, 1_000_000), (1000, 1000))
|
|
|
|
from mpi4py import MPI
|
|
from matrix_mpi import MatrixMPI
|
|
from vector_mpi import VectorMPI
|
|
|
|
comm = MPI.COMM_WORLD
|
|
rank = comm.Get_rank()
|
|
size = comm.Get_size()
|
|
|
|
m1 = MatrixMPI(list(range(1, 21)), (4, 5))
|
|
m2 = MatrixMPI(list(range(1, 16)), (5, 3))
|
|
|
|
m_mul = m1 * m2
|
|
|
|
v1 = VectorMPI(list(range(1, 21)))
|
|
v2 = VectorMPI(list(reversed(list(range(1, 21)))))
|
|
|
|
v_add = v1 + v2
|
|
v_mul = v1.T() * v2
|
|
|
|
if rank == 0:
|
|
print(m_mul)
|
|
print("---")
|
|
print(v_add)
|
|
print(v_mul)
|