Chen Zhang e20c92bb61 [Kernel] Move attn_type to Attention.__init__() (#11690) ...

Signed-off-by: Chen Zhang <zhangch99@outlook.com>

2025-01-07 00:11:28 +08:00

..

__init__.py

[Core] Refactor Attention Take 2 (#3462)

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

abstract.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

blocksparse_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

flash_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

flashinfer.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

hpu_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

ipex_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

openvino.py

[Model][OpenVINO] Fix regressions from #8346 (#10045)

2024-11-06 04:19:15 +00:00

pallas.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

placeholder_attn.py

[Doc] Update docs to refer to pooling models (#11093)

2024-12-11 13:36:27 +00:00

rocm_flash_attn.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

torch_sdpa.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00

utils.py

[torch.compile] support all attention backends (#10558)

2024-11-22 14:04:42 -08:00

xformers.py

[Kernel] Move attn_type to Attention.__init__() (#11690)

2025-01-07 00:11:28 +08:00