feat: reorganize casting utilities and enhance CPU support

[zhangyue207]

• Moved casting functions to separate CPU and CUDA headers for better organization.
• Introduced a new Cast() function in the CPU implementation to handle type conversions, including support for custom types like float16_t and bfloat16_t.
• Updated various operators to utilize the new casting utilities, ensuring consistent type handling across CPU and CUDA backends.
• Enhanced test cases to cover additional data types and ensure compatibility with the new casting logic.

test

pytest -k [cpu|cuda]

[zhangyue207]

(python3.10) zhangyue@server:~/InfiniOps$ pytest tests/ -k cuda
================================= test session starts ==================================
platform linux -- Python 3.10.19, pytest-9.0.2, pluggy-1.6.0
rootdir: /home/zhangyue/InfiniOps
configfile: pyproject.toml
plugins: xdist-3.8.0, cov-7.0.0
collected 3372 items / 1686 deselected / 1686 selected                                 

tests/test_add.py .............................................................. [  3%]
..............................................                                   [  6%]
tests/test_causal_softmax.py ..................                                  [  7%]
tests/test_gemm.py ............................................................. [ 11%]
................................................................................ [ 15%]
................................................................................ [ 20%]
................................................................................ [ 25%]
................................................................................ [ 30%]
................................................................................ [ 34%]
................................................................................ [ 39%]
................................................................................ [ 44%]
................................................................................ [ 49%]
................................................................................ [ 53%]
................................................................................ [ 58%]
................................................................................ [ 63%]
................................................................................ [ 68%]
................................................................................ [ 72%]
................................................................................ [ 77%]
................................................................................ [ 82%]
................................................................................ [ 87%]
................................................................................ [ 91%]
...............................................................................  [ 96%]
tests/test_rms_norm.py ....................................                      [ 98%]
tests/test_swiglu.py ........................                                    [100%]

======================== 1686 passed, 1686 deselected in 5.30s =========================

[zhangyue207]

(python3.10) zhangyue@server:~/InfiniOps$ pytest -k cpu
================================= test session starts ==================================
platform linux -- Python 3.10.19, pytest-9.0.2, pluggy-1.6.0
rootdir: /home/zhangyue/InfiniOps
configfile: pyproject.toml
testpaths: tests
plugins: xdist-3.8.0, cov-7.0.0
collected 3372 items / 1686 deselected / 1686 selected                                 

tests/test_add.py .............................................................. [  3%]
..............................................                                   [  6%]
tests/test_causal_softmax.py ..................                                  [  7%]
tests/test_gemm.py ............................................................. [ 11%]
................................................................................ [ 15%]
................................................................................ [ 20%]
................................................................................ [ 25%]
................................................................................ [ 30%]
................................................................................ [ 34%]
................................................................................ [ 39%]
................................................................................ [ 44%]
................................................................................ [ 49%]
................................................................................ [ 53%]
................................................................................ [ 58%]
................................................................................ [ 63%]
................................................................................ [ 68%]
................................................................................ [ 72%]
................................................................................ [ 77%]
................................................................................ [ 82%]
................................................................................ [ 87%]
................................................................................ [ 91%]
...............................................................................  [ 96%]
tests/test_rms_norm.py ....................................                      [ 98%]
tests/test_swiglu.py ........................                                    [100%]

================== 1686 passed, 1686 deselected in 243.35s (0:04:03) ===================

[zhangyue207]

root@iluvatar:/workspace/InfiniOps# pytest -k cpu
==================================================== test session starts ====================================================
platform linux -- Python 3.10.18, pytest-9.0.2, pluggy-1.6.0
rootdir: /workspace/InfiniOps
configfile: pyproject.toml
testpaths: tests
plugins: anyio-4.9.0, cov-7.0.0, xdist-3.8.0, typeguard-4.4.4
collected 3372 items / 1686 deselected / 1686 selected                                                                      

tests/test_add.py ................................................................................................... [  5%]
.........                                                                                                             [  6%]
tests/test_causal_softmax.py ..................                                                                       [  7%]
tests/test_gemm.py .................................................................................................. [ 13%]
..................................................................................................................... [ 20%]
..................................................................................................................... [ 27%]
..................................................................................................................... [ 34%]
..................................................................................................................... [ 41%]
..................................................................................................................... [ 47%]
..................................................................................................................... [ 54%]
..................................................................................................................... [ 61%]
..................................................................................................................... [ 68%]
..................................................................................................................... [ 75%]
..................................................................................................................... [ 82%]
..................................................................................................................... [ 89%]
...................................................................................................................   [ 96%]
tests/test_rms_norm.py ....................................                                                           [ 98%]
tests/test_swiglu.py ........................                                                                         [100%]

===================================== 1686 passed, 1686 deselected in 241.49s (0:04:01) =====================================

[zhangyue207]

root@iluvatar:/workspace/InfiniOps# ruff check . --exclude InfiniCore All checks passed!