Skip to content

issue/899 - fix: fix causal_softmax and rearrange bug#971

Open
spike-zhu wants to merge 1 commit intomainfrom
issue/899
Open

issue/899 - fix: fix causal_softmax and rearrange bug#971
spike-zhu wants to merge 1 commit intomainfrom
issue/899

Conversation

@spike-zhu
Copy link
Contributor

参考 nv 平台,修复摩尔平台的 rearrange 和 causal_softmax。

rearrange python 测试:
image

causal_softmax python 测试:
image

@spike-zhu spike-zhu requested a review from a team January 22, 2026 11:46
@spike-zhu spike-zhu self-assigned this Jan 22, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants