Pass block sizes to the second mm.arrangement in addmm.arrangement

[voltjia]

pytest output:

============================= test session starts ==============================
platform linux -- Python 3.10.12, pytest-8.3.3, pluggy-1.5.0
rootdir: /home/huangjiacheng/ntops
configfile: pyproject.toml
plugins: cov-6.0.0
collected 2 items

tests/test_addmm.py ..                                                   [100%]

============================== 2 passed in 16.91s ==============================