Merge pull request #2886 from devitocodes/tma-write #312

Job	Run time
pytest-mpi (pytest-mpi-docker-icx)	3m 47s
pytest-mpi (3.11)	45m 46s
pytest-mpi (3.10)	45m 43s
pytest-mpi (pytest-mpi-docker-gcc)	4m 26s
pytest-mpi (pytest-mpi-docker-gcc-arm)	4m 27s
	1h 44m 9s

Provide feedback