Driss Guessous 72c7ba484d Fix Hopper tests (#2242)

2026-02-08 09:25:01 -08:00

..

cute

Fix Hopper tests (#2242)

2026-02-08 09:25:01 -08:00

flash_attn_triton_amd

Add FLASH_ATTENTION_TRITON_AMD_CONFIG_JSON env var support (#2239)

2026-02-07 08:13:18 -08:00

layers

[Rotary] Don't need to wrap in custom_op, just need wrap_triton

2025-04-25 17:14:20 -04:00

losses

[CrossEntropy] Support precomputed LSE

2024-09-08 09:24:43 -07:00

models

minify torch.torch.int32 to torch.int32 (#1237)

2024-09-18 00:32:59 -07:00

modules

Fix import error

2025-04-25 14:48:27 -04:00

ops

[LayerNorm] Don't let torch.compile trace inside _layer_norm_bwd

2025-04-29 23:42:49 -04:00

utils

[Cute] Fix sink impl

2025-08-05 20:42:41 -04:00

__init__.py

Bump to v2.8.3

2025-08-14 13:11:47 -04:00

bert_padding.py

minify torch.torch.int32 to torch.int32 (#1237)

2024-09-18 00:32:59 -07:00

flash_attn_interface.py

[AMD] Triton Backend for ROCm #3 (#2178)

2026-01-28 07:49:08 -08:00

flash_attn_triton_og.py

Run isort and black on python files

2023-08-18 14:22:11 -07:00

flash_attn_triton.py

Run isort and black on python files

2023-08-18 14:22:11 -07:00

flash_blocksparse_attention.py

minor changes to unpad_input test util func

2024-09-16 14:24:11 -07:00

flash_blocksparse_attn_interface.py

Run isort and black on python files

2023-08-18 14:22:11 -07:00

pyproject.toml

Set line-length for ruff

2025-06-25 15:59:57 -04:00