Cyrus Leung 6ffa3f314c [CI/Build] Avoid CUDA initialization (#8534)

2024-09-18 10:38:11 +00:00

..

blocksparse_attention

[CI/Build] Avoid CUDA initialization (#8534)

2024-09-18 10:38:11 +00:00

__init__.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

ipex_attn.py

[Core/Bugfix] Add FP8 K/V Scale and dtype conversion for prefix/prefill Triton Kernel (#7208)

2024-08-12 22:47:41 +00:00

paged_attn.py

[Core/Bugfix] Add FP8 K/V Scale and dtype conversion for prefix/prefill Triton Kernel (#7208)

2024-08-12 22:47:41 +00:00

prefix_prefill.py

[CI/Build] Avoid CUDA initialization (#8534)

2024-09-18 10:38:11 +00:00

triton_flash_attention.py

[ROCm][AMD][Bugfix] adding a missing triton autotune config (#4845)

2024-05-16 10:46:52 -07:00