Skip to content

CUDA作业 - #60

Open
ShengnanLiao wants to merge 10 commits into
InfiniTensor:masterfrom
ShengnanLiao:master
Open

CUDA作业#60
ShengnanLiao wants to merge 10 commits into
InfiniTensor:masterfrom
ShengnanLiao:master

Conversation

@ShengnanLiao

Copy link
Copy Markdown

完成RMSnorm和FlashAttention的算子开发,在英伟达、天数、摩尔线程、沐曦卡上做了适配,全部pass。

Removed commented-out sections and explanations related to tensor layout and computation order in flashAttention function.
Added test results for rmsNorm and Attention tests, showing all verifications passed for both float and half data types.
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.

1 participant