[Cherry-pick release/1.1]Support more GPU arches for FA

[sneaxiy]

Before submitting

• Lint code. If there are lint issues, please format the code first.

# Install and register `pre-commit` in the project folder
pip install pre-commit && pre-commit install

# Process previous code files separately
pre-commit run --file XXXX.py

• Add test cases into tests folder. If there are codecov issues, please add tests cases first.

PR types

New features

PR changes

Others

Description

Support more GPU arches for FA.
Merged in dev: #4136

[paddle-bot]

Thanks for your contribution!

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.
⚠️ Please upload report for BASE (release/1.1@1f7612e). Learn more about missing BASE report.

Additional details and impacted files

@@              Coverage Diff               @@
##             release/1.1    #4137   +/-   ##
==============================================
  Coverage               ?   32.18%           
==============================================
  Files                  ?      435           
  Lines                  ?    82570           
  Branches               ?        0           
==============================================
  Hits                   ?    26575           
  Misses                 ?    55995           
  Partials               ?        0           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[From00]

LGTM