feat(cambricon): integrate InfiniOps backend - #1522
Merged
Merged
Conversation
baominghelly
force-pushed
the
feat/cambricon-infiniops
branch
2 times, most recently
from
August 24, 2026 04:48
bf18b87 to
a4337b4
Compare
baominghelly
marked this pull request as ready for review
August 24, 2026 06:03
voltjia
approved these changes
Aug 24, 2026
wooway777
approved these changes
Aug 24, 2026
wooway777
requested changes
Aug 24, 2026
wooway777
left a comment
Collaborator
There was a problem hiding this comment.
conflict了,另外xmake.lua额外需要这么多修改的理由是什么?有些东西是不是应该收束到bang.lua里面
baominghelly
force-pushed
the
feat/cambricon-infiniops
branch
from
August 24, 2026 07:44
a4337b4 to
38af5fe
Compare
Contributor
Author
conflict已解决。xmake.lua之前有一些清理之前编译缓存的操作,现在已删除;剩下的还有一个算子白名单的功能,因为寒武纪目前算子偏少,不加的话导入会报错算子找不到,这个功能等下一阶段算子补充好并且模型验证完后会去掉。 |
wooway777
approved these changes
Aug 25, 2026
voltjia
approved these changes
Aug 25, 2026
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
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.
Summary
INFINI_OPS_OPSadapters, PyTorch/torch_mlu ABI and link settings, and a single dispatch translation unitVerification
Tested in
infinicore-cambricon-dev-libaomingwith:export INFINI_OPS_OPS=add,argmax,copy,embedding,fused_add_rms_norm,gemm,rms_norm,rotary_embedding,silu_and_mul xmake build xmake install xmake build _infinicore xmake install _infinicoreThe complete build/install chain exited successfully. Both imports below succeeded with the generated extension:
Related InfiniOps operator PRs:
copyprovider InfiniOps#927rotary_embeddingprovider InfiniOps#944