Change PyPI name back to flash-attn-4

Author: tridao

.github/workflows/publish-fa4.yml

@@ -1,4 +1,4 @@
-name: Publish fa4 to PyPI
+name: Publish flash-attn-4 to PyPI
 
 on:
   push:
@@ -19,10 +19,13 @@ jobs:
       with:
         python-version: '3.12'
     - name: Install build dependencies
-      run: pip install build
+      run: pip install build twine
     - name: Build package
       run: python -m build
       working-directory: flash_attn/cute
+    - name: Check package metadata
+      run: twine check dist/*
+      working-directory: flash_attn/cute
     - name: Store distribution packages
       uses: actions/upload-artifact@v4
       with:
@@ -47,14 +50,16 @@ jobs:
   publish-to-pypi:
     needs: build
     runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/p/flash-attn-4
+    permissions:
+      id-token: write
     steps:
     - name: Download distribution packages
       uses: actions/download-artifact@v4
       with:
         name: python-package-distributions
         path: dist/
     - name: Publish to PyPI
-      env:
-        TWINE_USERNAME: "__token__"
-        TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}
-      run: pip install "twine>=6.1" "packaging>=24.2" && python -m twine upload dist/*
+      uses: pypa/gh-action-pypi-publish@release/v1

README.md

@@ -68,7 +68,7 @@ FlashAttention-4 is written in CuTeDSL and optimized for Hopper and Blackwell GP
 
 To install:
 ```sh
-pip install fa4
+pip install flash-attn-4
 ```
 
 Once installed, you can use it as follows:

flash_attn/cute/README.md

@@ -5,7 +5,7 @@ FlashAttention-4 is a CuTeDSL-based implementation of FlashAttention for Hopper
 ## Installation
 
 ```sh
-pip install fa4
+pip install flash-attn-4
 ```
 
 ## Usage

flash_attn/cute/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools>=75", "setuptools-scm>=8"]
 build-backend = "setuptools.build_meta"
 
 [project]
-name = "fa4"
+name = "flash-attn-4"
 dynamic = ["version"]
 description = "Flash Attention CUTE (CUDA Template Engine) implementation"
 readme = "README.md"