feat(ascend): add `causal_softmax` operator#780

Draft

zhangyue207 wants to merge 1 commit into

InfiniTensor:masterfrom

zhangyue207:feat/ascend-causal-softmax-operator

zhangyue207 commented Jun 30, 2026 •

edited

Loading

Collaborator

Summary

Add the Ascend causal_softmax implementation using ACLNN softmax/masked-fill primitives.
Keep the existing upstream base operator definition.
Update tests/test_causal_softmax.py coverage for Ascend.

Validation

git diff --check upstream/master...HEAD
clang-format --dry-run --Werror on changed C++ files
ruff format --check tests/test_causal_softmax.py
ruff check tests/test_causal_softmax.py


          feat(ascend): add causal_softmax operator

01bec69

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet