master

cann-robotchunkprefill_mask_pa

7d4c8401创建于 3 天前历史提交

文件	最后提交记录	最后更新时间
ops	chunkprefill_mask_pa Co-authored-by: lccchh<linchenghao@huawei.com> # message auto-generated for no-merge-commit merge: !700 merge 950_ChunkPrefill into master chunkprefill_mask_pa Created-by: lccchh Commit-by: lccchh Merged-by: cann-robot Description: ## 描述 新增适配Ascend950硬件的，不同于库上存量模板，使用[2048, 2048]压缩mask的flash attention，并支持以下特性：  ## 关联的Issue https://gitcode.com/cann/catlass/issues/234 ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!700	3 天前
__init__.py	chunkprefill_mask_pa Co-authored-by: lccchh<linchenghao@huawei.com> # message auto-generated for no-merge-commit merge: !700 merge 950_ChunkPrefill into master chunkprefill_mask_pa Created-by: lccchh Commit-by: lccchh Merged-by: cann-robot Description: ## 描述 新增适配Ascend950硬件的，不同于库上存量模板，使用[2048, 2048]压缩mask的flash attention，并支持以下特性：  ## 关联的Issue https://gitcode.com/cann/catlass/issues/234 ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!700	3 天前