Skip to content

[release/3.4][Build] Compile FlashAttention on arm64 platforms#79496

Open
risemeup1111 wants to merge 2 commits into
PaddlePaddle:release/3.4from
risemeup1111:cherry-pick/79488/release/3.4
Open

[release/3.4][Build] Compile FlashAttention on arm64 platforms#79496
risemeup1111 wants to merge 2 commits into
PaddlePaddle:release/3.4from
risemeup1111:cherry-pick/79488/release/3.4

Conversation

@risemeup1111

@risemeup1111 risemeup1111 commented Jul 17, 2026

Copy link
Copy Markdown
Contributor

PR Category

Environment Adaptation

PR Types

Improvements

Description

  1. Remove the NOT WITH_ARM guard from GPU third-party dependency conditions.
  2. Allow FlashAttention-related GPU third-party dependency logic to be enabled on ARM64 GPU builds.

Tests

  • git diff --check -- cmake/third_party.cmake
  • Planned: ARM64 + GPU CMake/build validation with WITH_ARM=ON and WITH_GPU=ON to verify FlashAttention-related third-party dependency logic is enabled on ARM64 GPU builds.

是否引起精度变化

否。本变更仅调整 ARM64 GPU 构建依赖开关,不修改数值计算逻辑或算子实现。


Cherry-pick of #79488 and #79500 (authored by @liuhao2638) to release/3.4.

devPR:#79488

---------

Co-authored-by: Claude <noreply@anthropic.com>
@paddle-bot

paddle-bot Bot commented Jul 17, 2026

Copy link
Copy Markdown

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

---------

Co-authored-by: risemeup1111 <13125134909@163.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

contributor External developers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants