Feat/add rmsnorm multi backend#55

Draft

gongchensu wants to merge 16 commits into

InfiniTensor:feat/ascend-operatorsfrom

gongchensu:feat/add-rmsnorm-multi-backend

Commits on Apr 15, 2026

fix(ascend): refine framework layer — caching, naming, build fixes
zhangyue
committed
feat(base): add new operator base classes and refine existing ones
zhangyue
committed
feat(cpu): add CPU implementations for Cast, Cat, Linear, Mul
zhangyue
committed
feat(ascend): add Ascend operator kernels for all operators
zhangyue
committed
feat(ascend): add multi-implementation variants and ATB operators
zhangyue
committed
feat(ascend): add custom AscendC kernels for RmsNorm and AddRmsNorm
zhangyue
committed
test(ascend): add comprehensive tests for all Ascend operators
zhangyue
committed
style: apply clang-format and fix code convention violations (round 1)
zhangyue
committed
style: fix code convention violations (round 2)
zhangyue
committed
style: fix code convention violations (round 3)
zhangyue
committed
fix(ci): add Ascend toolkit environment variables to CI Dockerfile
zhangyue
committed
style: apply ruff format and clang-format to all modified files
zhangyue
committed
style: fix ruff F401 lint errors for side-effect imports
zhangyue
committed
refactor(test): remove duplicate rms_norm test files and unify kernel loading
zhangyue
committed
feat(ops): add CPU and shared CUDA implementations for AddRmsNorm
gongchensu
committed
feat(ops): add NVIDIA, MetaX, Iluvatar, and Moore backends for AddRmsNorm
gongchensu
committed