Finish Assignment-Zane - #65
Open
Yzinuo wants to merge 14 commits into
Open
Conversation
added 14 commits
August 6, 2026 18:33
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
这次完成了 NVIDIA CUDA 和天数智芯 Iluvatar CoreX 两个平台的适配。
复现流程：
CPU / 通用构建
xmake
xmake install
pip install ./python
NVIDIA
xmake f --nv-gpu=y -c
xmake
xmake install
天数智芯 CoreX
export COREX_PATH=/usr/local/corex-4.4.0
export CUDA_PATH=$COREX_PATH
export LD_LIBRARY_PATH=$COREX_PATH/lib64:$LD_LIBRARY_PATH
xmake f -c --iluvatar-gpu=y --cuda=$COREX_PATH
xmake
xmake install
天数平台测试时使用：
python test/test_runtime.py --device iluvatar
python test/ops/add.py --device iluvatar
python test/ops/argmax.py --device iluvatar
python test/ops/embedding.py --device iluvatar
python test/ops/linear.py --device iluvatar
python test/ops/rms_norm.py --device iluvatar
python test/ops/rope.py --device iluvatar
python test/ops/self_attention.py --device iluvatar
python test/ops/swiglu.py --device iluvatar
复现结果：
CPU：GitHub Actions 的 Ubuntu 和 Windows 测试通过，Assignment 1、2、3 均正常。
NVIDIA CUDA：本地 Windows NVIDIA 环境已完成运行时、全部算子和模型推理一致性测试。
天数智芯 Iluvatar CoreX：在 BI-V150 服务器上完成干净构建，运行时测试和 8 个算子测试全部通过，包括大尺寸 RoPE 的 F32、F16、BF16 测试。
支持状态：
平台 状态 说明
CPU 已支持 Windows、Ubuntu CI 已验证
NVIDIA CUDA 已支持 Runtime、算子、模型推理已验证
天数智芯 Iluvatar CoreX / BI-V150 已支持 Runtime 与全部 Assignment 4 算子已在真实服务器验证