Add input precision support for matmul operations

[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 6 items

tests/test_addmm.py ..                                                   [ 33%]
tests/test_bmm.py ..                                                     [ 66%]
tests/test_mm.py ..                                                      [100%]

============================== 6 passed in 43.18s ==============================

[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 6 items

tests/test_addmm.py ..                                                   [ 33%]
tests/test_bmm.py ..                                                     [ 66%]
tests/test_mm.py ..                                                      [100%]

============================== 6 passed in 40.12s ==============================